Test estimator correction

ayrna · Nov 20, 2023 · dee720d · dee720d
1 parent 103021c
commit dee720d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/dlordinal/estimator/tests/test_estimator.py b/dlordinal/estimator/tests/test_estimator.py
@@ -38,7 +38,6 @@ def create_example_dataloader(batch_size, num_samples, input_size):
 
 
 def calculate_loss(model, loss_fn, dataloader):
-    losses = []
     model.eval()
     total_loss = 0.0