Fix lint errors

fennel-ai · Sep 21, 2024 · 402874f · 402874f
1 parent fb751c1
commit 402874f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/fennel/client_tests/test_struct_type.py b/fennel/client_tests/test_struct_type.py
@@ -280,7 +280,9 @@ def test_struct_type_with_firstk(client):
     assert (
         len(df["MovieFeaturesWithFirstK.cast_list"][0]) == 2
     )  # 2 cast members for "Inception"
-    assert len(df["MovieFeaturesWithFirstK.cast_list"][1]) == 2  # 2 cast members for
+    assert (
+        len(df["MovieFeaturesWithFirstK.cast_list"][1]) == 2
+    )  # 2 cast members for
     # "Titanic"
     ellen = Cast(name="Ellen Page", actor_id=2, age=34)
     cast1 = df["MovieFeaturesWithFirstK.cast_list"][0][0]

diff --git a/fennel/datasets/test_dataset.py b/fennel/datasets/test_dataset.py
@@ -310,7 +310,16 @@ def create_aggregated_dataset(cls, user_info: Dataset):
                     {"name": "count", "dtype": {"intType": {}}},
                     {"name": "avg_age", "dtype": {"doubleType": {}}},
                     {"name": "stddev_age", "dtype": {"doubleType": {}}},
-                    {"name": "countries", "dtype": {"arrayType": {"of": {"optionalType": {"of": {"stringType": {}}}}}}},
+                    {
+                        "name": "countries",
+                        "dtype": {
+                            "arrayType": {
+                                "of": {
+                                    "optionalType": {"of": {"stringType": {}}}
+                                }
+                            }
+                        },
+                    },
                 ]
             },
             "timestamp": "timestamp",