Resolve typing error

lanl · Sep 27, 2024 · 3dd11d7 · 3dd11d7
1 parent cf360bd
commit 3dd11d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scico/flax/train/checkpoints.py b/scico/flax/train/checkpoints.py
@@ -89,7 +89,7 @@ def checkpoint_save(state: TrainState, config: ConfigDict, workdir: Union[str, P
         # Remove non-serializable partial functools in post_lst if it exists
         config_ = config.copy()
         if "post_lst" in config_:
-            config_.pop("post_lst", None)
+            config_.pop("post_lst", None)  # type: ignore
         mngr.save(
             step,
             args=ocp.args.Composite(