fix(tasks): update ticks on loss plot axes

Signed-off-by: Cameron Smith <[email protected]>

src/pyrovelocity/tasks/train.py

@@ -9,6 +9,7 @@
 import numpy as np
 from anndata._core.anndata import AnnData
 from beartype import beartype
+from matplotlib.ticker import ScalarFormatter, SymmetricalLogLocator
 from mlflow import MlflowClient
 from numpy import ndarray
 from scvi.model._utils import parse_device_args
@@ -416,6 +417,13 @@ def set_loss_plot_axes(ax):
     ax.set_xlabel("Epochs")
     ax.set_ylabel("-ELBO")
 
+    locator = SymmetricalLogLocator(base=10, linthresh=1, subs=[1.0])
+    ax.yaxis.set_major_locator(locator)
+
+    formatter = ScalarFormatter()
+    formatter.set_scientific(False)
+    ax.yaxis.set_major_formatter(formatter)
+
 
 def log_run_info(r: mlflow.entities.run.Run) -> None:
     tags = {k: v for k, v in r.data.tags.items() if not k.startswith("mlflow.")}