File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 66
77class Settings (BaseSettings ):
88 data_path : Path = Path (os .getenv ("DATAPATH" , "/arccnet/data" ))
9- models_path : Path = Path (os .getenv ("MODESLPATH " , "/arccnet/models" ))
9+ models_path : Path = Path (os .getenv ("MODELSPATH " , "/arccnet/models" ))
1010
1111
1212settings = Settings ()
Original file line number Diff line number Diff line change @@ -17,8 +17,8 @@ def download_and_extract_model(
1717 model_url ,
1818 model_name ,
1919 * ,
20- extracted_weights_filename = "model-data/comet-torch-model.pth" ,
2120 model_data_path ,
21+ extracted_weights_filename = "model-data/comet-torch-model.pth" ,
2222):
2323 """
2424 Downloads and extracts a model archive.
@@ -85,7 +85,7 @@ def download_and_extract_model(
8585 # If extracted file has different name, create expected path
8686 actual_extracted_path = model_cache_dir / target_file
8787 if actual_extracted_path != extracted_path :
88- actual_extracted_path .move (extracted_path )
88+ actual_extracted_path .replace (extracted_path )
8989
9090 if not extracted_path .exists ():
9191 raise FileNotFoundError (f"Weights file not found at { extracted_path } " )
You can’t perform that action at this time.
0 commit comments