Merge pull request #3 from code-kern-ai/update-build-script

Updates Container Name

Author: FelixKirsch

build

@@ -1,3 +1,12 @@
 #!/bin/bash
+IS_ARM64=""
+currentArch="$(uname -m)"
+if [ "$currentArch" == "arm64" ];
+then
+    echo "architecture: arm64"
+    IS_ARM64="_arm64"
+else
+    echo "architecture: $currentArch"
+fi
 
-docker build -t registry.dev.onetask.ai/code-kern-ai/ml-execution-environment:main .
+docker build -t registry.dev.onetask.ai/code-kern-ai/refinery-ml-exec-env:dev$IS_ARM64 .

requirements.txt

@@ -20,5 +20,6 @@ six==1.16.0
 threadpoolctl==3.1.0
 tinycss2==1.1.1
 tomli==2.0.1
+torch==1.11.0
 typing_extensions==4.2.0
 urllib3==1.26.9