Updated the prediction method for udocker

leschultz · Feb 13, 2024 · 47e0286 · 47e0286
1 parent da74d6a
commit 47e0286
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 # Package information
 name = 'madml'
-version = '2.3.8'  # Need to increment every time to push to PyPI
+version = '2.3.9'  # Need to increment every time to push to PyPI
 description = 'Application domain of machine learning in materials science.'
 url = 'https://github.com/leschultz/'\
       'materials_application_domain_machine_learning.git'

diff --git a/src/madml/hosting/docker.py b/src/madml/hosting/docker.py
@@ -49,6 +49,7 @@ def predict(self, df):
         command = 'udocker --allow-root run -v '
         command += '{}:/mnt '.format(os.getcwd())
         command += self.container
+        command += ' python3 model_predict.py'
 
         subprocess.check_output(
                                 command,