Formatting

Author: osyuksel

pymc/model_graph.py

@@ -293,9 +293,7 @@ def vars_to_plot(self, var_names: Iterable[str] | None = None) -> list[str]:
         # ordering of self._all_var_names is important
         return [get_var_name(var) for var in selected_ancestors]
 
-    def make_compute_graph(
-        self, var_names: Iterable[str] | None = None
-    ) -> dict[str, set[str]]:
+    def make_compute_graph(self, var_names: Iterable[str] | None = None) -> dict[str, set[str]]:
         """Get map of var_name -> set(input var names) for the model."""
         model = self.model
         named_vars = self._all_vars

pymc/util.py

@@ -18,7 +18,7 @@
 from collections import namedtuple
 from collections.abc import Sequence
 from copy import deepcopy
-from typing import NewType, cast
+from typing import cast
 
 import arviz
 import cloudpickle
@@ -32,7 +32,6 @@
 from pymc.exceptions import BlockModelAccessError
 
 
-
 class _UnsetType:
     """Type for the `UNSET` object to make it look nice in `help(...)` outputs."""