Apply Black 20.8b1 formatting

This commit is contained in:
Seth Michael Larson 2020-08-27 15:19:56 -05:00 committed by GitHub
parent 1d6311164e
commit 1a8a301cd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 24 additions and 16 deletions

View File

@ -464,14 +464,14 @@ class FieldMappings:
-------------------------------------------- --------------------------------------------
``` ```
Pandas dtype Python type NumPy type Usage Pandas dtype Python type NumPy type Usage
object str string_, unicode_ Text object str string_, unicode_ Text
int64 int int_, int8, int16, int32, int64, uint8, uint16, uint32, uint64 Integer numbers int64 int int_, int8, int16, int32, int64, uint8, uint16, uint32, uint64 Integer numbers
float64 float float_, float16, float32, float64 Floating point numbers float64 float float_, float16, float32, float64 Floating point numbers
bool bool bool_ True/False values bool bool bool_ True/False values
datetime64 NA datetime64[ns] Date and time values datetime64 NA datetime64[ns] Date and time values
timedelta[ns] NA NA Differences between two datetimes timedelta[ns] NA NA Differences between two datetimes
category NA NA Finite list of text values category NA NA Finite list of text values
``` ```
""" """
es_dtype = None es_dtype = None

View File

@ -218,7 +218,8 @@ class ImportedMLModel(MLModel):
body["definition"] = serializer.serialize_model() body["definition"] = serializer.serialize_model()
self._client.ml.put_trained_model( self._client.ml.put_trained_model(
model_id=self._model_id, body=body, model_id=self._model_id,
body=body,
) )
def predict(self, X: Union[List[float], List[List[float]]]) -> np.ndarray: def predict(self, X: Union[List[float], List[List[float]]]) -> np.ndarray:

View File

@ -93,7 +93,8 @@ class LGBMForestTransformer(ModelTransformer):
self, tree_id: int, node_id: int, tree_node_json_obj: Dict[str, Any] self, tree_id: int, node_id: int, tree_node_json_obj: Dict[str, Any]
) -> TreeNode: ) -> TreeNode:
return TreeNode( return TreeNode(
node_idx=node_id, leaf_value=[float(tree_node_json_obj["leaf_value"])], node_idx=node_id,
leaf_value=[float(tree_node_json_obj["leaf_value"])],
) )
def build_tree(self, tree_id: int, tree_json_obj: Dict[str, Any]) -> Tree: def build_tree(self, tree_id: int, tree_json_obj: Dict[str, Any]) -> Tree:

View File

@ -137,7 +137,12 @@ class Query:
self._aggs[name] = agg self._aggs[name] = agg
def hist_aggs( def hist_aggs(
self, name: str, field: str, min_value: Any, max_value: Any, num_bins: int, self,
name: str,
field: str,
min_value: Any,
max_value: Any,
num_bins: int,
) -> None: ) -> None:
""" """
Add histogram agg e.g. Add histogram agg e.g.

View File

@ -30,7 +30,8 @@ ELASTICSEARCH_HOST = os.environ.get("ELASTICSEARCH_HOST") or "localhost"
TEST_SUITE = os.environ.get("TEST_SUITE", "xpack") TEST_SUITE = os.environ.get("TEST_SUITE", "xpack")
if TEST_SUITE == "xpack": if TEST_SUITE == "xpack":
ES_TEST_CLIENT = Elasticsearch( ES_TEST_CLIENT = Elasticsearch(
ELASTICSEARCH_HOST, http_auth=("elastic", "changeme"), ELASTICSEARCH_HOST,
http_auth=("elastic", "changeme"),
) )
else: else:
ES_TEST_CLIENT = Elasticsearch(ELASTICSEARCH_HOST) ES_TEST_CLIENT = Elasticsearch(ELASTICSEARCH_HOST)

View File

@ -36,7 +36,7 @@ class TestDataFrameDateTime(TestData):
@classmethod @classmethod
def setup_class(cls): def setup_class(cls):
""" setup any state specific to the execution of the given class (which """setup any state specific to the execution of the given class (which
usually contains tests). usually contains tests).
""" """
es = ES_TEST_CLIENT es = ES_TEST_CLIENT
@ -65,7 +65,7 @@ class TestDataFrameDateTime(TestData):
@classmethod @classmethod
def teardown_class(cls): def teardown_class(cls):
""" teardown any state that was previously setup with a call to """teardown any state that was previously setup with a call to
setup_class. setup_class.
""" """

View File

@ -29,7 +29,7 @@ class TestDateTime(TestData):
@classmethod @classmethod
def setup_class(cls): def setup_class(cls):
""" setup any state specific to the execution of the given class (which """setup any state specific to the execution of the given class (which
usually contains tests). usually contains tests).
""" """
es = ES_TEST_CLIENT es = ES_TEST_CLIENT
@ -58,7 +58,7 @@ class TestDateTime(TestData):
@classmethod @classmethod
def teardown_class(cls): def teardown_class(cls):
""" teardown any state that was previously setup with a call to """teardown any state that was previously setup with a call to
setup_class. setup_class.
""" """