Skip to content

Commit b6f460a

Browse files
authored
Merge pull request #13 from predict-idlab/12-no_kind_in_core
Remove kind parameter from Core client
2 parents 0cb1e10 + 4480c12 commit b6f460a

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

src/obelisk/asynchronous/core.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@
1919
from typing_extensions import Self
2020
from numbers import Number
2121

22+
from obelisk.strategies.retry import NoRetryStrategy, RetryStrategy
23+
from obelisk.types import ObeliskKind
24+
2225

2326
DataType = Literal['number', 'number[]', 'json', 'bool', 'string']
2427
"""The possible types of data Obelisk can accept"""
@@ -180,6 +183,16 @@ class Client(BaseClient):
180183
page_limit: int = 250
181184
"""How many datapoints to request per page in a cursored fetch"""
182185

186+
187+
def __init__(self, client: str, secret: str,
188+
retry_strategy: RetryStrategy = NoRetryStrategy()) -> None:
189+
BaseClient.__init__(
190+
client=client,
191+
secret=secret,
192+
retry_strategy=retry_strategy,
193+
kind=ObeliskKind.CORE
194+
)
195+
183196
async def send(
184197
self,
185198
dataset: str,

0 commit comments

Comments
 (0)