Fixed bug in len

jeffrey-hokanson · May 28, 2019 · 228e5e6 · 228e5e6
1 parent 648f752
commit 228e5e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psdr/polyridge.py b/psdr/polyridge.py
@@ -305,7 +305,7 @@ def __init__(self, degree, subspace_dimension, basis = 'legendre',
 		self.bound = bound
 
 	def __len__(self):
-		return U.shape[0]
+		return self.U.shape[0]
 
 
 	def fit(self, X, fX, U0 = None, **kwargs):