diff --git a/monai/auto3dseg/algo_gen.py b/monai/auto3dseg/algo_gen.py index 37a94712cb..ad185117a4 100644 --- a/monai/auto3dseg/algo_gen.py +++ b/monai/auto3dseg/algo_gen.py @@ -9,7 +9,6 @@ # See the License for the specific language governing permissions and # limitations under the License. - from monai.transforms import Randomizable diff --git a/monai/handlers/mlflow_handler.py b/monai/handlers/mlflow_handler.py index 277903e8e2..a06f2013aa 100644 --- a/monai/handlers/mlflow_handler.py +++ b/monai/handlers/mlflow_handler.py @@ -280,7 +280,7 @@ def _default_iteration_log(self, engine: Engine) -> None: # If there is optimizer attr in engine, then record parameters specified in init function. if hasattr(engine, "optimizer"): - cur_optimizer = engine.optimizer # type: ignore + cur_optimizer = engine.optimizer for param_name in self.optimizer_param_names: params = { f"{param_name} group_{i}": float(param_group[param_name])