Lint

tests/integration_tests/utils.py

@@ -6,12 +6,12 @@
  AvgAggregationOp,
  CountAggregationOp,
  ExistsAggregationOp,
+ FirstAggregationOp,
+ LastAggregationOp,
  MajorityAggregationOp,
  MaxAggregationOp,
  MinAggregationOp,
  SumAggregationOp,
- FirstAggregationOp,
- LastAggregationOp,
 )
 from trane.ops.filter_ops import (
  AllFilterOp,

tests/ops/test_aggregation_ops.py

@@ -5,12 +5,12 @@
 from trane.ops.aggregation_ops import (
  AvgAggregationOp,
  CountAggregationOp,
+ FirstAggregationOp,
+ LastAggregationOp,
  MajorityAggregationOp,
  MaxAggregationOp,
  MinAggregationOp,
  SumAggregationOp,
- FirstAggregationOp,
- LastAggregationOp
 )
 from trane.ops.utils import get_aggregation_ops
 
@@ -56,12 +56,14 @@ def test_min_agg_op(df):
  assert output == np.min(df["col"])
  assert "the minimum <col> in all related records" in op.generate_description()
 
+
 def test_first_agg_op(df):
  op = FirstAggregationOp("col")
  output = op(df)
  assert output == df["col"].iloc[0]
  assert "the first <col> in all related records" in op.generate_description()
 
+
 def test_last_agg_op(df):
  op = LastAggregationOp("col")
  output = op(df)

tests/ops/test_op_utils.py

@@ -3,12 +3,12 @@
  AvgAggregationOp,
  CountAggregationOp,
  ExistsAggregationOp,
+ FirstAggregationOp,
+ LastAggregationOp,
  MajorityAggregationOp,
  MaxAggregationOp,
  MinAggregationOp,
  SumAggregationOp,
- FirstAggregationOp,
- LastAggregationOp
 )
 from trane.ops.filter_ops import (
  AllFilterOp,

tests/test_utils.py

@@ -14,12 +14,12 @@
  AvgAggregationOp,
  CountAggregationOp,
  ExistsAggregationOp,
+ FirstAggregationOp,
+ LastAggregationOp,
  MajorityAggregationOp,
  MaxAggregationOp,
  MinAggregationOp,
  SumAggregationOp,
- FirstAggregationOp,
- LastAggregationOp
 )
 from trane.ops.filter_ops import (
  AllFilterOp,
@@ -249,6 +249,7 @@ def test_check_operations_cat():
  result, modified_meta = _check_operations_valid(operations, table_meta)
  assert result is True
 
+
 def test_foreign_key():
  table_meta = {
  "id": ("Categorical", {"primary_key", "category"}),

trane/ops/aggregation_ops.py

@@ -149,6 +149,7 @@ def label_function(self, dataslice):
  return None
  return dataslice[self.column_name].iloc[0]
 
+
 class LastAggregationOp(AggregationOpBase):
  input_output_types = [("category", "category")]
  description = " the last <{}> in all related records"