Update model publication code

Closes #355 (closed)

added P2 label

assigned to @mblanco

$ export ARKINDEX_API_URL=https://ee.preprod.arkindex.teklia.com
$ export ARKINDEX_API_TOKEN=<token>
$ export ARKINDEX_WORKER_RUN_ID=88ed2f8d-bc7e-4594-8b9a-a2f246cfcef6

• Model available

from pathlib import Path

from arkindex_worker.worker import DatasetWorker
from arkindex_worker.worker.training import create_archive, TrainingMixin

class MyWorker(DatasetWorker, TrainingMixin):
    pass

worker = MyWorker()
worker.create_model_version("84eea59f-3f1f-4193-a579-147d1b6c04ee")

with create_archive(Path("LaPlusBelle")) as (archive_path, hash, size, archive_hash):
    worker.upload_to_s3(archive_path)
    worker.validate_model_version(hash, size, archive_hash)

• Model already available

2024-06-14 09:34:07,953 INFO/arkindex_worker: Uploading to s3...
2024-06-14 09:34:10,317 WARNING/arkindex_worker: An available model version exists with hash 464d95fdbeb6d1a9f6a4c10d84cdcd56, using it instead of the pending version.
2024-06-14 09:34:10,317 WARNING/arkindex_worker: Removing the pending model version.
2024-06-14 09:34:10,413 INFO/arkindex_worker: Retrieving the existing model version.
2024-06-14 09:34:10,500 INFO/arkindex_worker: Model version 5437ff49-0c43-42e9-96c5-a988e97880df is now available.

added 1 commit

• 693c2714 - Retrieve existing model

Compare with previous version

added 4 commits

• 693c2714...79996b63 - 2 commits from branch master
• 9c244a7a - Update model publication code
• be6a42de - Retrieve existing model

Compare with previous version

requested review from @yschneider

requested changes

resolved all threads

added 1 commit

• 17215479 - Apply suggestions

Compare with previous version

mentioned in issue yolo#45 (closed)

mentioned in issue pylaia#66 (closed)

requested review from @yschneider

approved this merge request

merged