add JVM memory input

KRR-Oxford · Feb 10, 2023 · de20f01 · de20f01
1 parent b32b1b1
commit de20f01
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/deeponto/__init__.py b/src/deeponto/__init__.py
@@ -31,4 +31,5 @@ def init_jvm(memory):
             convertStrings=False)
 
     if jpype.isJVMStarted():
+        print(f"{memory} maximum memory allocated to JVM.")
         print("JVM started successfully.")
diff --git a/src/deeponto/onto/ontology.py b/src/deeponto/onto/ontology.py
@@ -26,7 +26,10 @@
 from deeponto import init_jvm
 
 # initialise JVM for python-java interaction
-init_jvm("2g")
+import click
+memory = click.prompt("Please enter the maximum memory located to JVM", type=str, default="8g")
+print()
+init_jvm(memory)
 
 from java.io import File  # type: ignore
 from java.util import Collections  # type: ignore