Fix tests

sbrunk · Feb 5, 2024 · 8e05618 · 8e05618
1 parent 11eb92c
commit 8e05618
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/core/src/test/scala/TrainingSuite.scala b/core/src/test/scala/TrainingSuite.scala
@@ -57,11 +57,11 @@ class TraininSuite extends munit.FunSuite {
         noGrad {
           weight.grad.foreach { grad =>
             weight -= grad * learningRate
-            grad.zero()
+            grad.zero_()
           }
           bias.grad.foreach { grad =>
             weight -= grad * learningRate
-            grad.zero()
+            grad.zero_()
           }
         }
         loss

diff --git a/core/src/test/scala/torch/DeviceSuite.scala b/core/src/test/scala/torch/DeviceSuite.scala
@@ -17,15 +17,9 @@
 package torch
 
 import munit.ScalaCheckSuite
-import torch.DeviceType.CUDA
 import org.scalacheck.Prop.*
-import org.bytedeco.pytorch.global.torch as torch_native
-import org.scalacheck.{Arbitrary, Gen}
 import org.scalacheck._
-import Gen._
-import Arbitrary.arbitrary
-import DeviceType.CPU
-import Generators.{*, given}
+import Generators.given
 
 class DeviceSuite extends ScalaCheckSuite {
   test("device native roundtrip") {