Upload script.py

script.py

@@ -101,7 +101,28 @@ def convert_entry_to_human_readable(entry):
 '''---end of compulsory---'''
 
 ### The part below is used to define and test your solution.
-
+import subprocess
+import sys
+import os
+def setup_environment():
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "git+http://hf.co/usm3d/tools.git"])
+    import hoho
+    hoho.setup()
+    
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.0.1", "torchvision==0.15.2", "torchaudio==2.0.2", "-f", "https://download.pytorch.org/whl/cu117.html"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "scikit-learn"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "tqdm"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "scipy"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "open3d"])
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "easydict"])
+
+    pc_util_path = os.path.join(os.getcwd(), 'pc_util')
+    if os.path.isdir(pc_util_path):
+        os.chdir(pc_util_path)
+        subprocess.check_call([sys.executable, "setup.py", "install"])
+    else:
+        print(f"Directory {pc_util_path} does not exist")
+    
 from pathlib import Path
 def save_submission(submission, path):
     """
@@ -116,6 +137,8 @@ def save_submission(submission, path):
     print(f"Submission saved to {path}")
 
 if __name__ == "__main__":
+	setup_environment()
+
     from handcrafted_solution import predict
     print ("------------ Loading dataset------------ ")
     params = hoho.get_params()