fix python wrapper.

donaldlab · Nov 29, 2018 · d0aa7fd · d0aa7fd
1 parent 18e9d84
commit d0aa7fd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/osprey/__init__.py b/python/osprey/__init__.py
@@ -1331,11 +1331,11 @@ def MSKStar(objective, constraints=[], epsilon=useJavaDefault, objectiveWindowSi
 	return builder.build()
 
 def PartitionFunctionFactory(confSpace, confEcalc, state, confUpperBoundcalc=None):
-    pfuncFactory = c.kstar.pfunc.PartitionFunctionFactory(confSpace, state)
+    pfuncFactory = c.kstar.pfunc.PartitionFunctionFactory(confSpace, confEcalc, state)
     if confUpperBoundcalc is not None:
-        pfuncFactory.setUseMARKStar(confUpperBoundcalc,confEcalc)
+        pfuncFactory.setUseMARKStar(confUpperBoundcalc)
     else:
-        pfuncFactory.setUseGradientDescent(confEcalc)
+        pfuncFactory.setUseGradientDescent()
 
     return pfuncFactory