ruff

Author: xadupre

pandas_streaming/df/connex_split.py

@@ -61,6 +61,7 @@ def train_test_split_weights(
                 f"test_size={test_size} or train_size={train_size} cannot be null (1)."
             )
         from sklearn.model_selection import train_test_split
+
         return train_test_split(
             df, test_size=test_size, train_size=train_size, random_state=random_state
         )

pandas_streaming/df/dataframe_helpers.py

@@ -155,13 +155,13 @@ def hash_floatl(c):
     }  # pylint: disable=R1721
     for c in cols:
         t = coltype[c]
-        if t == int: # noqa: E721
+        if t == int:  # noqa: E721
             df[c] = df[c].apply(hash_intl)
         elif t == numpy.int64:
             df[c] = df[c].apply(lambda x: numpy.int64(hash_intl(x)))
-        elif t == float: # noqa: E721
+        elif t == float:  # noqa: E721
             df[c] = df[c].apply(hash_floatl)
-        elif t == object: # noqa: E721
+        elif t == object:  # noqa: E721
             df[c] = df[c].apply(hash_strl)
         else:
             raise NotImplementedError(  # pragma: no cover