Merge pull request #43 from ruivieira/FAI-705

Add post-install command to get Java dependencies

Author: tteofili

.github/workflows/workflow.yml

@@ -28,10 +28,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt
           pip install -r requirements-dev.txt
-      - name: Download JARs
-        run: |
-          chmod +x ./deps.sh
-          ./deps.sh
+          python setup.py build install
       - name: Lint
         run: |
           pylint --ignore-imports=yes $(find trustyai -type f -name "*.py")

setup.py

@@ -1,10 +1,30 @@
 import os
 from setuptools import setup
+from setuptools.command.install import install
+import site
 
 here = os.path.abspath(os.path.dirname(__file__))
 with open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
     long_description = f.read()
 
+TRUSTY_VERSION = "1.12.0.Final"
+
+
+class PostInstall(install):
+    """Fetch TrustyAI explainability JARs from Maven Central"""
+
+    def run(self):
+        install.run(self)
+        _ROOT = os.path.join(site.getsitepackages()[0], "trustyai", "dep")
+        print(f"Installing Maven dependencies into {_ROOT}")
+        os.system(f"mvn org.apache.maven.plugins:maven-dependency-plugin:2.10:get "
+                  f"-DremoteRepositories=https://repository.sonatype.org/content/repositories/central  "
+                  f"-Dartifact=org.kie.kogito:explainability-core:{TRUSTY_VERSION} -Dmaven.repo.local={_ROOT} -q")
+        _TESTS_FILE = os.path.join("org", "kie", "kogito", "explainability-core", TRUSTY_VERSION,
+                                   f"explainability-core-{TRUSTY_VERSION}-tests.jar")
+        os.system(f"wget -O {os.path.join(_ROOT, _TESTS_FILE)} https://repo1.maven.org/maven2/{_TESTS_FILE}")
+
+
 setup(
     name="trustyai",
     version="0.0.5",
@@ -29,4 +49,5 @@
     packages=['trustyai', 'trustyai.model', 'trustyai.utils', 'trustyai.local'],
     include_package_data=False,
     install_requires=['Jpype1'],
+    cmdclass={"install": PostInstall},
 )

tests/common.py

@@ -12,15 +12,7 @@
 INITIALISED = False
 
 if not INITIALISED:
-    trustyai.init(
-        path=trustyai.CORE_DEPS + [
-            "./dep/org/optaplanner/optaplanner-core/8.12.0.Final/optaplanner-core-8.12.0.Final.jar",
-            "./dep/org/apache/commons/commons-math3/3.6.1/commons-math3-3.6.1.jar",
-            "./dep/org/kie/kie-api/7.59.0.Final/kie-api-7.59.0.Final.jar",
-            "./dep/io/micrometer/micrometer-core/1.7.4/micrometer-core-1.7.4.jar",
-        ]
-    )
-
+    trustyai.init()
     INITIALISED = True
 
 from trustyai.model import (

trustyai/__init__.py

@@ -1,16 +1,25 @@
 # pylint: disable = import-error, import-outside-toplevel, dangerous-default-value, invalid-name, R0801
 """Main TrustyAI Python bindings"""
 from typing import List
+import site
+import os
 import uuid
 import jpype
 import jpype.imports
 from jpype import _jcustomizer, _jclass
 
 TRUSTY_VERSION = "1.12.0.Final"
+DEFAULT_DEP_PATH = os.path.join(site.getsitepackages()[0], "trustyai", "dep")
+
 CORE_DEPS = [
-    f"./dep/org/kie/kogito/explainability-core/{TRUSTY_VERSION}/*",
-    "./dep/org/slf4j/slf4j-api/1.7.30/slf4j-api-1.7.30.jar",
-    "./dep/org/apache/commons/commons-lang3/3.12.0/commons-lang3-3.12.0.jar",
+    f"{DEFAULT_DEP_PATH}/org/kie/kogito/explainability-core/{TRUSTY_VERSION}/*",
+    f"{DEFAULT_DEP_PATH}/org/slf4j/slf4j-api/1.7.30/slf4j-api-1.7.30.jar",
+    f"{DEFAULT_DEP_PATH}/org/apache/commons/commons-lang3/3.12.0/commons-lang3-3.12.0.jar",
+    f"{DEFAULT_DEP_PATH}/org/optaplanner/optaplanner-core/8.12.0.Final/"
+    f"optaplanner-core-8.12.0.Final.jar",
+    f"{DEFAULT_DEP_PATH}/org/apache/commons/commons-math3/3.6.1/commons-math3-3.6.1.jar",
+    f"{DEFAULT_DEP_PATH}/org/kie/kie-api/7.59.0.Final/kie-api-7.59.0.Final.jar",
+    f"{DEFAULT_DEP_PATH}/io/micrometer/micrometer-core/1.7.4/micrometer-core-1.7.4.jar",
 ]