We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5712459 commit 9fd567fCopy full SHA for 9fd567f
tests/tests_pytorch/checkpointing/test_model_checkpoint.py
@@ -29,10 +29,10 @@
29
import pytest
30
import torch
31
import yaml
32
-from jsonargparse import ArgumentParser
33
from lightning.fabric.utilities.cloud_io import _load as pl_load
34
from lightning.pytorch import Trainer, seed_everything
35
from lightning.pytorch.callbacks import ModelCheckpoint
+from lightning.pytorch.cli import LightningArgumentParser as ArgumentParser
36
from lightning.pytorch.demos.boring_classes import BoringModel
37
from lightning.pytorch.loggers import CSVLogger, TensorBoardLogger
38
from lightning.pytorch.utilities.exceptions import MisconfigurationException
0 commit comments