Skip to content

Commit

Permalink
Cleaned up code
Browse files Browse the repository at this point in the history
  • Loading branch information
vruusmann committed Dec 3, 2024
1 parent ff37897 commit 53012b3
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions jpmml_evaluator/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -228,16 +228,17 @@ def setLocatable(self, locatable = False):
return self

def loadFile(self, path):
file = self.backend.newObject("java.io.File", path)
self.javaModelEvaluatorBuilder.load(file)
javaFile = self.backend.newObject("java.io.File", path)
self.javaModelEvaluatorBuilder.load(javaFile)
return self

def loadString(self, string):
return self.loadBytes(bytes(string, "utf-8"))

def loadBytes(self, bytes):
stream = self.backend.newObject("java.io.ByteArrayInputStream", bytes)
self.javaModelEvaluatorBuilder.load(stream)
javaInputStream = self.backend.newObject("java.io.ByteArrayInputStream", bytes)
self.javaModelEvaluatorBuilder.load(javaInputStream)
javaInputStream.close()
return self

def transform(self, javaPMMLTransformer):
Expand All @@ -246,12 +247,12 @@ def transform(self, javaPMMLTransformer):

def transpile(self, path = None):
if path:
file = self.backend.newObject("java.io.File", path)
transpiler = self.backend.newObject("org.jpmml.transpiler.FileTranspiler", None, file)
javaFile = self.backend.newObject("java.io.File", path)
javaTranspiler = self.backend.newObject("org.jpmml.transpiler.FileTranspiler", None, javaFile)
else:
transpiler = self.backend.newObject("org.jpmml.transpiler.InMemoryTranspiler", None)
transformer = self.backend.newObject("org.jpmml.transpiler.TranspilerTransformer", transpiler)
return self.transform(transformer)
javaTranspiler = self.backend.newObject("org.jpmml.transpiler.InMemoryTranspiler", None)
javaTranspilerTransformer = self.backend.newObject("org.jpmml.transpiler.TranspilerTransformer", javaTranspiler)
return self.transform(javaTranspilerTransformer)

def make_backend(alias):
if not isinstance(alias, str):
Expand Down

0 comments on commit 53012b3

Please sign in to comment.