universal title and ruff fix

Author: Arina Danilina

src/moscot/plotting/_plotting.py

@@ -402,7 +402,7 @@ def pull(
 
     fig = _plot_scatter(
         adata=adata,
-        generic_key=data["key"] if "key" in data else None,
+        generic_key=data["key"],
         key_stored=key,
         source=data["source"],
         target=data["target"],

src/moscot/plotting/_utils.py

@@ -429,14 +429,10 @@ def _plot_scatter(
             titles = [f"{categories if categories is not None else 'Cells'} at time {source if push else target}"]
             titles.extend([f"{name} at time {time_points[i]}" for i in range(1, len(time_points))])
         else:
-            titles = (
-                [f"{categories if categories is not None else 'Cells'} at time {source if push else target} and {name}"]
-                if generic_key is None
-                else [
+            titles = ([
                     f"{'Push' if push else 'Pull'} {categories if categories is not None else 'cells'} "
                     + f"from {source if push else target} to {target if push else source}"
-                ]
-            )
+                ])
     for i, ax in enumerate(axs):
         # we need to create adata_view because otherwise the view of the adata is copied in the next step i+1
         with RandomKeys(adata, n=2, where="obs") as keys: