fix formatting

Signed-off-by: kalyan <[email protected]>
rawwar · Oct 13, 2023 · 00a3dbe · 00a3dbe
1 parent da9f744
commit 00a3dbe
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/opensearch_py_ml/field_mappings.py b/opensearch_py_ml/field_mappings.py
@@ -91,7 +91,9 @@ def is_numeric(self) -> bool:
 
     @property
     def is_timestamp(self) -> bool:
-        return is_datetime64_any_dtype(self.pd_dtype) or is_timedelta64_dtype(self.pd_dtype)
+        return is_datetime64_any_dtype(self.pd_dtype) or is_timedelta64_dtype(
+            self.pd_dtype
+        )
 
     @property
     def is_bool(self) -> bool:
@@ -794,7 +796,9 @@ def metric_source_fields(
                 pd_dtypes.append(np.dtype(pd_dtype))
                 os_field_names.append(os_field_name)
                 os_date_formats.append(os_date_format)
-            elif include_timestamp and (is_datetime64_any_dtype(pd_dtype) or is_timedelta64_dtype(pd_dtype)):
+            elif include_timestamp and (
+                is_datetime64_any_dtype(pd_dtype) or is_timedelta64_dtype(pd_dtype)
+            ):
                 pd_dtypes.append(np.dtype(pd_dtype))
                 os_field_names.append(os_field_name)
                 os_date_formats.append(os_date_format)