formatting files

start.py

@@ -6,7 +6,9 @@ from huggingface_hub import hf_hub_download
 
 def run_command(command: str, cwd: str = None) -> tuple:
     """Run a shell command in the specified directory and return the output."""
-    process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd)
+    process = subprocess.Popen(
+        command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd
+    )
     stdout, stderr = process.communicate()
     if process.returncode != 0:
         print(f"Error: {stderr.decode()}")
@@ -17,10 +19,12 @@ def run_command(command: str, cwd: str = None) -> tuple:
 
 def download_dataset():
     """Download the dataset."""
-    print("Downloading the dataset...")
+    print("Downloading the autocast dataset...")
     repo_id = "valory/autocast"
     base_dir = os.getcwd()
-    output_dir = os.path.join(base_dir, "olas-predict-benchmark", "benchmark", "data", "autocast")
+    output_dir = os.path.join(
+        base_dir, "olas-predict-benchmark", "benchmark", "data", "autocast"
+    )
     if not os.path.exists(output_dir):
         os.makedirs(output_dir, exist_ok=True)
     filenames = [
@@ -28,7 +32,12 @@ def download_dataset():
         "autocast_questions_filtered.pkl",
     ]
     for filename in filenames:
-        hf_hub_download(repo_id=repo_id, filename=filename, local_dir=output_dir, repo_type="dataset")
+        hf_hub_download(
+            repo_id=repo_id,
+            filename=filename,
+            local_dir=output_dir,
+            repo_type="dataset",
+        )
     print("Dataset downloaded successfully.")
 
 
@@ -43,7 +52,10 @@ def start():
         ("git submodule init", base_dir),
         ("git submodule update --init --recursive", base_dir),
         ("git submodule update --remote --recursive", base_dir),
-        ('git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*"', olas_dir),
+        (
+            'git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*"',
+            olas_dir,
+        ),
         ("git remote update", olas_dir),
         ("git fetch --all", olas_dir),
         ("git checkout main", olas_dir),
@@ -66,4 +78,4 @@ def start():
     download_dataset()
 
 
-start()
+start()