Fix building

build.gradle

@@ -155,7 +155,8 @@ ext {
 
     pythonProjects = [
       project(':h2o-py'),
-      project(':h2o-py-cloud-extensions')
+      project(':h2o-py-cloud-extensions'),
+      project(':h2o-py-mlflow-flavors')
     ]
 
     // The project which need to be run under CI only

h2o-py-mlflow-flavors/build.gradle

@@ -20,11 +20,11 @@ ext {
 //
 task createVersionFiles() {
     doLast {
-        file("${buildDir}/h2o/mlflow/").mkdirs()
-        File version_file = new File("${buildDir}/h2o-mlflow-flavors/", "version.txt")
+        file("${buildDir}/h2o_mlflow_flavors/").mkdirs()
+        File version_file = new File("${buildDir}/h2o_mlflow_flavors/", "version.txt")
         version_file.write(PROJECT_VERSION)
 
-        File build_file = new File("${buildDir}/h2o-mlflow-flavors/", "buildinfo.txt")
+        File build_file = new File("${buildDir}/h2o_mlflow_flavors/", "buildinfo.txt")
         build_file.write(buildVersion.toString())
     }
 }
@@ -34,7 +34,7 @@ task copySrcFiles(type: Copy) {
     from ("${projectDir}") {
         include "setup.py"
         include "setup.cfg"
-        include "h2o-mlflow-flavors/**"
+        include "h2o_mlflow_flavors/**"
         include "README.md"
         include "DESCRIPTION.rst"
     }
@@ -45,7 +45,7 @@ task buildDist(type: Exec, dependsOn: [createVersionFiles, copySrcFiles]) {
     workingDir buildDir
     doFirst {
         file("${buildDir}/tmp").mkdirs()
-        standardOutput = new FileOutputStream(file("${buildDir}/tmp/h2o-mlflow-flavors_buildDist.out"))
+        standardOutput = new FileOutputStream(file("${buildDir}/tmp/h2o_mlflow_flavors_buildDist.out"))
     }
     commandLine getOsSpecificCommandLine([pythonexe, "setup.py", "bdist_wheel"])
 }

h2o-py-mlflow-flavors/setup.cfg

@@ -12,7 +12,7 @@
 #
 ignore = E241,E265,E302,E303,E701,D105
 max-line-length = 120
-application-import-names = h2o-mlflow-flavors
+application-import-names = h2o_mlflow_flavors
 import-order-style = smarkets
 inline-quotes = "
 

h2o-py-mlflow-flavors/setup.py

@@ -77,5 +77,5 @@
     ]},
 
     # run-time dependencies
-    install_requires=["h2o-mlops-client>=0.58"]
+    install_requires=["mlflow>=1.29.0"]
 )

settings.gradle

@@ -10,6 +10,7 @@ include 'h2o-app'
 include 'h2o-r'
 include 'h2o-py'
 include 'h2o-py-cloud-extensions'
+include 'h2o-py-mlflow-flavors'
 include 'h2o-assemblies:main'
 include 'h2o-assemblies:minimal'
 include 'h2o-assemblies:steam'