Sadjad Alikhani commited on
Commit
506b17c
·
verified ·
1 Parent(s): f3de1b7

Delete lwm_setup.py

Browse files
Files changed (1) hide show
  1. lwm_setup.py +0 -51
lwm_setup.py DELETED
@@ -1,51 +0,0 @@
1
- import os
2
- import sys
3
- import subprocess
4
- import importlib.util
5
-
6
- class LWMSetup:
7
- def __init__(self, repo_url="https://huggingface.co/sadjadalikhani/LWM", clone_dir="./LWM"):
8
- self.repo_url = repo_url
9
- self.clone_dir = clone_dir
10
-
11
- def setup(self):
12
- # Step 1: Clone the public repository
13
- self.clone_repo()
14
-
15
- # Step 2: Add the cloned directory to the Python path so all imports work
16
- sys.path.append(self.clone_dir)
17
-
18
- # Step 3: Dynamically import all required modules
19
- self.import_modules()
20
-
21
- def clone_repo(self):
22
- if not os.path.exists(self.clone_dir):
23
- print(f"Cloning repository from {self.repo_url} into {self.clone_dir}")
24
-
25
- # Clone the repository without needing an authentication token
26
- result = subprocess.run(["git", "clone", self.repo_url, self.clone_dir], capture_output=True, text=True)
27
-
28
- if result.returncode != 0:
29
- print(f"Error cloning the repository: {result.stderr}")
30
- sys.exit(1) # Exit if cloning fails
31
- else:
32
- print(f"Repository cloned successfully into {self.clone_dir}")
33
- else:
34
- print(f"Repository already cloned into {self.clone_dir}")
35
-
36
- def import_modules(self):
37
- def import_module_from_file(module_name, file_path):
38
- spec = importlib.util.spec_from_file_location(module_name, file_path)
39
- module = importlib.util.module_from_spec(spec)
40
- spec.loader.exec_module(module)
41
- return module
42
-
43
- try:
44
- self.lwm_model = import_module_from_file("lwm_model", os.path.join(self.clone_dir, "lwm_model.py"))
45
- self.inference = import_module_from_file("inference", os.path.join(self.clone_dir, "inference.py"))
46
- self.load_data = import_module_from_file("load_data", os.path.join(self.clone_dir, "load_data.py"))
47
- self.input_preprocess = import_module_from_file("input_preprocess", os.path.join(self.clone_dir, "input_preprocess.py"))
48
- print("Modules imported successfully.")
49
- except ModuleNotFoundError as e:
50
- print(f"Error importing modules: {e}")
51
- sys.exit(1)