Skip to content

Commit b2e5b94

Browse files
authored
Fix typing
1 parent 339e9a9 commit b2e5b94

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/axiomatic/magic.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from IPython import get_ipython # type: ignore
22
from IPython.core.magic import register_line_cell_magic, register_line_magic # type: ignore
33
from IPython.display import HTML, display # type: ignore
4+
from typing import Optional
45

56
import platformdirs # type: ignore
67
import os
@@ -16,7 +17,7 @@ class AXMagic:
1617
client: Axiomatic
1718
query: str
1819

19-
def __init__(self, api_key: str = os.getenv("AXIOMATIC_API_KEY")):
20+
def __init__(self, api_key: Optional[str] = os.getenv("AXIOMATIC_API_KEY")):
2021
self.folder = platformdirs.user_config_dir("axiomatic")
2122

2223
self.api_key = api_key

0 commit comments

Comments
 (0)