Skip to content

Commit afb4f9c

Browse files
author
Sunil Thaha
committed
feat(model): enable loading models from local filesystem
Signed-off-by: Sunil Thaha <[email protected]>
1 parent 95ad4d2 commit afb4f9c

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ dependencies = [
4444
"boto3==1.34.155",
4545
"pymarkdownlnt==0.9.22",
4646
"yamllint==1.35.1",
47+
"requests-file==2.1.0",
4748
]
4849

4950
[project.scripts]
@@ -67,8 +68,6 @@ extra-dependencies = [
6768
"coverage[toml]>=6.5",
6869
"ipdb",
6970
"ipython",
70-
"ipdb",
71-
"ipython",
7271
"pytest",
7372
]
7473

src/kepler_model/estimate/archived_model.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,11 +71,16 @@ def get_achived_model(power_request):
7171
if url == "":
7272
print("no URL set for ", output_type_name, power_request.energy_source)
7373
return None
74-
print(f"try getting archieved model from URL: {url} for {output_type_name}")
75-
response = requests.get(url)
76-
print(response)
74+
logger.info(f"try getting archieved model from URL: {url} for {output_type_name}")
75+
76+
s = requests.Session()
77+
s.mount("file://", FileAdapter())
78+
response = s.get(url)
79+
logger.debug(f"response: {response}")
80+
7781
if response.status_code != 200:
7882
return None
83+
7984
output_path = unpack(power_request.energy_source, output_type, response, replace=False)
8085
if output_path is not None:
8186
metadata = load_metadata(output_path)

0 commit comments

Comments
 (0)