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 4b0c994 commit 67212c1Copy full SHA for 67212c1
src/maxdiffusion/tests/wan_checkpointer_test.py
@@ -249,14 +249,6 @@ def setUp(self):
249
self.config.dataset_type = "test_dataset"
250
251
252
-class WanCheckpointerEdgeCasesTest(unittest.TestCase):
253
- """Tests for edge cases and error handling."""
254
-
255
- def setUp(self):
256
- self.config = MagicMock()
257
- self.config.checkpoint_dir = "/tmp/wan_checkpoint_edge_test"
258
- self.config.dataset_type = "test_dataset"
259
260
@patch("maxdiffusion.checkpointing.wan_checkpointer.create_orbax_checkpoint_manager")
261
@patch("maxdiffusion.checkpointing.wan_checkpointer.WanPipeline2_1")
262
def test_load_checkpoint_with_explicit_none_step(self, mock_wan_pipeline, mock_create_manager):
0 commit comments