summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorTheodore Aptekarev <aptekarev@gmail.com>2023-09-20 10:46:23 +0200
committerTheodore Aptekarev <aptekarev@gmail.com>2023-09-20 10:46:23 +0200
commitac5bcc5e9a7fd8fd87369828d10fb610cb2ec302 (patch)
treeeb2682727d0276e88c4dcc6ca1a6b64b5582656f
parent6ed837d24c117b09b68f6fd84522886fa8c62bea (diff)
Fix linting errors
-rw-r--r--openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py30
1 files changed, 13 insertions, 17 deletions
diff --git a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py
index e71e4afb35c..1abb5f23853 100644
--- a/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py
+++ b/openbb_sdk/sdk/provider/openbb_provider/abstract/fetcher.py
@@ -17,9 +17,11 @@ class classproperty:
"""Class property decorator."""
def __init__(self, f):
+ """Initialize decorator."""
self.f = f
def __get__(self, obj, owner):
+ """Get the property."""
return self.f(owner)
@@ -105,18 +107,16 @@ class Fetcher(Generic[Q, R]):
# Query Assertions
assert query
assert issubclass(type(query), cls.query_params_type)
- assert all([getattr(query, key) == value for key, value in test_params.items()])
+ assert all(getattr(query, key) == value for key, value in test_params.items())
# Data Assertions
assert data
- is_list = type(data) == list
+ is_list = isinstance(data, list)
if is_list:
assert all(
- [
- field in data[0]
- for field in cls.data_type.__fields__
- if field in data[0]
- ]
+ field in data[0]
+ for field in cls.data_type.__fields__
+ if field in data[0]
)
# This makes sure that the data is not transformed yet so that the
# pipeline is implemented correctly. We can remove this assertion if we
@@ -124,7 +124,7 @@ class Fetcher(Generic[Q, R]):
assert issubclass(type(data[0]), cls.data_type) is False
else:
assert all(
- [field in data for field in cls.data_type.__fields__ if field in data]
+ field in data for field in cls.data_type.__fields__ if field in data
)
assert issubclass(type(data), cls.data_type) is False
@@ -133,14 +133,12 @@ class Fetcher(Generic[Q, R]):
# Transformed Data Assertions
assert transformed_data
- is_list = type(transformed_data) == list
+ is_list = isinstance(transformed_data, list)
if is_list:
assert len(transformed_data) > 0 # type: ignore
assert all(
- [
- field in transformed_data[0].__dict__ # type: ignore
- for field in cls.return_type.__args__[0].__fields__
- ]
+ field in transformed_data[0].__dict__ # type: ignore
+ for field in cls.return_type.__args__[0].__fields__
)
assert issubclass(type(transformed_data[0]), cls.data_type) # type: ignore
assert issubclass(
@@ -149,10 +147,8 @@ class Fetcher(Generic[Q, R]):
)
else:
assert all(
- [
- field in transformed_data.__dict__
- for field in cls.return_type.__fields__
- ]
+ field in transformed_data.__dict__
+ for field in cls.return_type.__fields__
)
assert issubclass(type(transformed_data), cls.data_type)
assert issubclass(type(transformed_data), cls.return_type)