Skip to content

Commit f928d57

Browse files
committed
Use cloudnet-api-client in model processing
1 parent e871947 commit f928d57

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/processing/model.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,14 +44,21 @@ def process_model(processor: Processor, params: ModelParams, directory: Path):
4444
start_date = params.date
4545
end_date = params.date
4646

47-
upload_meta = processor.get_model_upload(params, start_date, end_date)
47+
upload_meta = processor.client.raw_model_metadata(
48+
params.site.id,
49+
model_id=params.model.id,
50+
date_from=start_date,
51+
date_to=end_date,
52+
)
53+
4854
if not upload_meta:
4955
msg = "No valid model upload found"
5056
raise SkipTaskError(msg)
5157

5258
raw_dir = directory / "raw"
5359
raw_dir.mkdir()
54-
full_paths, raw_uuids = processor.download_raw_data(upload_meta, raw_dir)
60+
full_paths = processor.client.download(upload_meta, raw_dir, progress=False)
61+
raw_uuids = [raw.uuid for raw in upload_meta]
5562

5663
volatile = True
5764
if existing_meta := processor.get_model_file(params):

0 commit comments

Comments
 (0)