Modernise

great_ai/tracing/add_ground_truth.py

@@ -23,7 +23,7 @@ def add_ground_truth(
     tags: Union[List[str], str] = [],
     train_split_ratio: float = 1,
     test_split_ratio: float = 0,
-    validation_split_ratio: float = 0
+    validation_split_ratio: float = 0,
 ) -> None:
     """Add training data (with optional train-test splitting).
 

great_ai/tracing/query_ground_truth.py

@@ -10,7 +10,7 @@ def query_ground_truth(
     *,
     since: Optional[datetime] = None,
     until: Optional[datetime] = None,
-    return_max_count: Optional[int] = None
+    return_max_count: Optional[int] = None,
 ) -> List[Trace]:
     """Return training samples.
 

great_ai/tracing/tracing_context.py

@@ -51,15 +51,19 @@ class TracingContext(Generic[T]):
                     logged_values=self._values,
                     models=self._models,
                     output=output,
-                    exception=None
-                    if exception is None
-                    else f"{type(exception).__name__}: {exception}",
+                    exception=(
+                        None
+                        if exception is None
+                        else f"{type(exception).__name__}: {exception}"
+                    ),
                     tags=[
                         self._name,
                         ONLINE_TAG_NAME,
-                        PRODUCTION_TAG_NAME
-                        if get_context().is_production
-                        else DEVELOPMENT_TAG_NAME,
+                        (
+                            PRODUCTION_TAG_NAME
+                            if get_context().is_production
+                            else DEVELOPMENT_TAG_NAME
+                        ),
                     ],
                 ),
             )