Skip to content

Commit

Permalink
fix failing test
Browse files Browse the repository at this point in the history
  • Loading branch information
nnaakkaaii committed Jul 14, 2024
1 parent f644551 commit 37abb19
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/models/test_basic_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def test_basic_model():
)
dataloader = DataLoader(FakeDataset(), batch_size=4)

model = BasicModel(network, optimizer, scheduler, criterion)
model = BasicModel(network, "", optimizer, scheduler, criterion)
with TemporaryDirectory() as tempdir:
model.train(
dataloader,
Expand Down

0 comments on commit 37abb19

Please sign in to comment.