Use correct registry

ekut-es · Aug 23, 2024 · e008a5f · e008a5f
1 parent 222eadd
commit e008a5f
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/.github/workflows/publish_images.yml b/.github/workflows/publish_images.yml
@@ -88,7 +88,7 @@ jobs:
         with:
           context: .
           push: true
-          tags: ${{ env.image_name }}:latest
+          tags: ${{ env.REGISTRY }}/${{ env.image_name }}:latest
           build-args: |
             python_version=${{ matrix.python_version }}
             target=${{ matrix.target }}

diff --git a/Dockerfile b/Dockerfile
@@ -39,7 +39,6 @@ RUN  if [ "$target" = "hannah" ] || [ "$target" = "mlonmcu" ]; then\
         apt-get update -y && apt-get -y install git mesa-utils  libblas-dev liblapack-dev libsndfile1-dev libsox-dev cmake ninja-build curl build-essential;  \
       fi
 
-RUN echo "Target: $target"
 
 # Install poetry using recommended method
 RUN  pip install poetry

diff --git a/hannah/train.py b/hannah/train.py
@@ -120,9 +120,6 @@ def train(
         logger.append(CSVLogger(".", version="logs", name=""))
         logger.append(JSONLogger(".", version="logs", name=""))
 
-        # if DVCLIVE_AVAILABLE:
-        #    logger.append(DVCLogger())
-
         callbacks = []
 
         callbacks.extend(list(common_callbacks(config)))