This might fix the problems for now. Will improve later.

gugarosa · Jan 15, 2020 · 43fa985 · 43fa985
1 parent af260ff
commit 43fa985
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/opfython/core/node.py b/opfython/core/node.py
@@ -157,7 +157,7 @@ def cost(self):
 
     @cost.setter
     def cost(self, cost):
-        if not (isinstance(cost, float) or isinstance(cost, int) or isinstance(cost, np.int64) or isinstance(cost, np.int64)):
+        if not (isinstance(cost, float) or isinstance(cost, int) or isinstance(cost, np.int32) or isinstance(cost, np.int64)):
             raise e.TypeError('`cost` should be a float or integer')
 
         self._cost = cost
@@ -172,7 +172,7 @@ def density(self):
 
     @density.setter
     def density(self, density):
-        if not (isinstance(density, float) or isinstance(density, int) or isinstance(density, np.int32)):
+        if not (isinstance(density, float) or isinstance(density, int) or isinstance(density, np.int32) or isinstance(density, np.int64)):
             raise e.TypeError('`density` should be a float or integer')
 
         self._density = density
@@ -187,7 +187,7 @@ def radius(self):
 
     @radius.setter
     def radius(self, radius):
-        if not (isinstance(radius, float) or isinstance(radius, int) or isinstance(radius, np.int32)):
+        if not (isinstance(radius, float) or isinstance(radius, int) or isinstance(radius, np.int32) or isinstance(radius, np.int64)):
             raise e.TypeError('`radius` should be a float or integer')
 
         self._radius = radius

diff --git a/opfython/subgraphs/knn.py b/opfython/subgraphs/knn.py
@@ -88,7 +88,7 @@ def constant(self):
 
     @constant.setter
     def constant(self, constant):
-        if not (isinstance(constant, float) or isinstance(constant, int) or isinstance(constant, np.int32)):
+        if not (isinstance(constant, float) or isinstance(constant, int) or isinstance(constant, np.int32) or isinstance(constant, np.int64)):
             raise e.TypeError('`constant` should be a float or integer')
 
         self._constant = constant
@@ -103,7 +103,7 @@ def density(self):
 
     @density.setter
     def density(self, density):
-        if not (isinstance(density, float) or isinstance(density, int) or isinstance(density, np.int32)):
+        if not (isinstance(density, float) or isinstance(density, int) or isinstance(density, np.int32) or isinstance(density, np.int64)):
             raise e.TypeError('`density` should be a float or integer')
 
         self._density = density
@@ -118,7 +118,7 @@ def min_density(self):
 
     @min_density.setter
     def min_density(self, min_density):
-        if not (isinstance(min_density, float) or isinstance(min_density, int) or isinstance(min_density, np.int32)):
+        if not (isinstance(min_density, float) or isinstance(min_density, int) or isinstance(min_density, np.int32) or isinstance(min_density, np.int64)):
             raise e.TypeError('`min_density` should be a float or integer')
 
         self._min_density = min_density
@@ -133,7 +133,7 @@ def max_density(self):
 
     @max_density.setter
     def max_density(self, max_density):
-        if not (isinstance(max_density, float) or isinstance(max_density, int) or isinstance(max_density, np.int32)):
+        if not (isinstance(max_density, float) or isinstance(max_density, int) or isinstance(max_density, np.int32) or isinstance(max_density, np.int64)):
             raise e.TypeError('`max_density` should be a float or integer')
 
         self._max_density = max_density