Skip to content

Commit 872688d

Browse files
committed
Fix circular import
1 parent 2daa7b7 commit 872688d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tabpfn_client/prompt_agent.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
import textwrap
22
import getpass
33
from password_strength import PasswordPolicy
4+
from typing import TYPE_CHECKING
45

5-
from tabpfn_client.service_wrapper import UserAuthenticationClient
6+
if TYPE_CHECKING:
7+
from tabpfn_client.tabpfn_classifier import UserAuthenticationClient
68

79

810
class PromptAgent:
@@ -41,7 +43,7 @@ def prompt_welcome(cls):
4143
print(cls.indent(prompt))
4244

4345
@classmethod
44-
def prompt_and_set_token(cls, user_auth_handler: UserAuthenticationClient):
46+
def prompt_and_set_token(cls, user_auth_handler: "UserAuthenticationClient"):
4547
# Choose between registration and login
4648
prompt = "\n".join(
4749
[

0 commit comments

Comments
 (0)