Skip to content

Commit 02a1395

Browse files
committed
feat(updating-auto-fs-creation): ignoring mypy
1 parent 2307444 commit 02a1395

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

butterfree/automated/feature_set_creation.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,6 @@ class Table: # noqa: D101
3333
class FeatureSetCreation:
3434
"""Class to auto-generate readers and features."""
3535

36-
def __init__(self):
37-
pass
38-
3936
def _get_features_with_regex(self, sql_query: str) -> List[str]:
4037
features = []
4138
sql_query = " ".join(sql_query.split())
@@ -67,12 +64,13 @@ def _get_data_type(self, field_name: str, df: pd.DataFrame) -> str:
6764
if "type" in field_type_keys and "elementType" in field_type_keys:
6865
return (
6966
"."
70-
+ BUTTERFREE_DTYPES[field_type["type"]][
67+
+ BUTTERFREE_DTYPES[field_type["type"]][ # type: ignore
7168
field_type["elementType"]
7269
]
7370
)
7471

7572
return "." + BUTTERFREE_DTYPES[field["type"]]
73+
7674
return ""
7775

7876
def _get_tables_with_regex(self, sql_query: str) -> Tuple[List[Table], str]:
@@ -184,7 +182,7 @@ def get_features(self, sql_query: str, df: Optional[DataFrame] = None) -> str:
184182
data_type = "."
185183

186184
if df is None:
187-
df = spark.sql(sql_query) # noqa: F821
185+
df = spark.sql(sql_query) # type: ignore # noqa: F821
188186

189187
data_type = self._get_data_type(feature, df)
190188

0 commit comments

Comments
 (0)