Merge branch 'montana/readme' of github.com:postgresml/postgresml into montana/readme

Montana Low · Montana Low · commit 4ca1a5f45432 · 2022-04-14T11:11:39.000-07:00
diff --git a/pgml/pgml/model.py b/pgml/pgml/model.py
@@ -214,7 +214,8 @@ def data(self):
             split = self.test_size
             if isinstance(split, float):
                 split = int(self.test_size * X.len())
-            return X[0:split], X[split:X.len()-1], y[0:split], y[split:y.len()-1]
+            return X[:split], X[split:], y[:split], y[split:]
+
 
         # TODO normalize and clean data
 
