From b9be04a51d6094511d89414be81c70bd47e3aacc Mon Sep 17 00:00:00 2001
From: Marek Novotny <marek.novotny@h2o.ai>
Date: Mon, 13 Nov 2023 19:18:53 +0100
Subject: [PATCH] Update build definition

---
 h2o-py-mlflow-flavor/build.gradle | 17 +----------------
 h2o-py-mlflow-flavor/setup.py     | 10 +---------
 2 files changed, 2 insertions(+), 25 deletions(-)

diff --git a/h2o-py-mlflow-flavor/build.gradle b/h2o-py-mlflow-flavor/build.gradle
index 58dcd1d243d2..2ba20ba9eb6f 100644
--- a/h2o-py-mlflow-flavor/build.gradle
+++ b/h2o-py-mlflow-flavor/build.gradle
@@ -15,21 +15,6 @@ ext {
     testsPath = file("tests")
 }
 
-//
-// Create a file with version for Python dist task
-//
-task createVersionFiles() {
-    doLast {
-        file("${buildDir}/h2o_mlflow_flavor/").mkdirs()
-        File version_file = new File("${buildDir}/h2o_mlflow_flavor/", "version.txt")
-        version_file.write(PROJECT_VERSION)
-
-        File build_file = new File("${buildDir}/h2o_mlflow_flavor/", "buildinfo.txt")
-        build_file.write(buildVersion.toString())
-    }
-}
-
-
 task copySrcFiles(type: Copy) {
     from ("${projectDir}") {
         include "setup.py"
@@ -41,7 +26,7 @@ task copySrcFiles(type: Copy) {
     into "${buildDir}"
 }
 
-task buildDist(type: Exec, dependsOn: [createVersionFiles, copySrcFiles]) {
+task buildDist(type: Exec, dependsOn: [copySrcFiles]) {
     workingDir buildDir
     doFirst {
         file("${buildDir}/tmp").mkdirs()
diff --git a/h2o-py-mlflow-flavor/setup.py b/h2o-py-mlflow-flavor/setup.py
index aeb84e7c8d50..5730e35ed4e7 100644
--- a/h2o-py-mlflow-flavor/setup.py
+++ b/h2o-py-mlflow-flavor/setup.py
@@ -9,11 +9,7 @@
 with open(os.path.join(here, 'DESCRIPTION.rst'), encoding='utf-8') as f:
     long_description = f.read()
 
-version = "0.1.0-SNAPSHOT"
-# Get the version from the relevant file
-with open(os.path.join(here, 'h2o_mlflow_flavor/version.txt'), encoding='utf-8') as f:
-    version = f.read()
-
+version = "0.1.0"
 packages = find_packages(exclude=["tests*"])
 print("Found packages: %r" % packages)
 
@@ -69,10 +65,6 @@
     keywords='ML Flow, H2O-3',
 
     packages=packages,
-    package_data={"h2o": [
-        "version.txt",      # version file
-        "buildinfo.txt"     # buildinfo file
-    ]},
 
     # run-time dependencies
     install_requires=["mlflow>=1.29.0"]