Lint

stanford-crfm · Jan 29, 2025 · 51bb0c0 · 51bb0c0
1 parent 4f9d414
commit 51bb0c0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/helm/benchmark/run_specs/sql_run_specs.py b/src/helm/benchmark/run_specs/sql_run_specs.py
@@ -23,6 +23,7 @@ def get_bird_sql_dev_run_spec() -> RunSpec:
         groups=["bird_sql"],
     )
 
+
 @run_spec_function("spider")
 def get_spider_run_spec() -> RunSpec:
     scenario_spec = ScenarioSpec(class_name="helm.benchmark.scenarios.spider_scenario.SpiderScenario")

diff --git a/src/helm/benchmark/scenarios/spider_scenario.py b/src/helm/benchmark/scenarios/spider_scenario.py
@@ -43,7 +43,10 @@ class SpiderScenario(Scenario):
     def get_instances(self, output_path: str) -> List[Instance]:
         data_parent_path = os.path.join(output_path, "data")
         ensure_file_downloaded(
-            "https://drive.google.com/uc?id=1403EGqzIDoHMdQF4c9Bkyl7dZLZ5Wt6J&export=download&confirm=t", data_parent_path, unpack=True, unpack_type="unzip"
+            "https://drive.google.com/uc?id=1403EGqzIDoHMdQF4c9Bkyl7dZLZ5Wt6J&export=download&confirm=t",
+            data_parent_path,
+            unpack=True,
+            unpack_type="unzip",
         )
         data_root_path = os.path.join(data_parent_path, "spider_data")
         # databases_unzip_target = os.path.join(data_root_path, "unzipped_dev_databases")