Fix version check and quick_test.py

PriorLabs · Mar 11, 2024 · bce902d · bce902d
1 parent c3e53e1
commit bce902d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/quick_test.py b/quick_test.py
@@ -34,8 +34,7 @@
         tabpfn_classifier.init()
         tabpfn = TabPFNClassifier(model="latest_tabpfn_hosted")
         # print("checking estimator", check_estimator(tabpfn))
-        print(X_train.shape[0]*100)
-        tabpfn.fit(np.repeat(X_train, 100, axis=0), np.repeat(y_train, 100, axis=0))
+        tabpfn.fit(X_train[:99], y_train[:99])
         print("predicting")
         print(tabpfn.predict(X_test))
         print("predicting_proba")

diff --git a/tabpfn_client/client.py b/tabpfn_client/client.py
@@ -278,7 +278,7 @@ def login(self, email: str, password: str) -> str | None:
             data=common_utils.to_oauth_request_form(email, password)
         )
 
-        self._validate_response(response, "login", only_version_check=True)
+        self._validate_response(response, "login", only_version_check=False)
         if response.status_code == 200:
             access_token = response.json()["access_token"]