Fix tests

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

Hide file tree

Showing 4 changed files with 6 additions and 15 deletions.
diff --git a/core/src/test/scala/TrainingSuite.scala b/core/src/test/scala/TrainingSuite.scala
@@ -28,8 +28,8 @@ class TraininSuite extends munit.FunSuite {
 
     torch.manualSeed(1)
 
-    var weight = torch.randn(Seq(1), requiresGrad = true)
-    var bias = torch.zeros(Seq(1), requiresGrad = true)
+    val weight = torch.randn(Seq(1), requiresGrad = true)
+    val bias = torch.zeros(Seq(1), requiresGrad = true)
 
     def model(xb: Tensor[Float32]): Tensor[Float32] = (xb matmul weight) + bias
 
@@ -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") {

diff --git a/core/src/test/scala/torch/TensorCheckSuite.scala b/core/src/test/scala/torch/TensorCheckSuite.scala
@@ -19,7 +19,7 @@ package torch
 import munit.ScalaCheckSuite
 import shapeless3.typeable.{TypeCase, Typeable}
 import shapeless3.typeable.syntax.typeable.*
-import Generators.{*, given}
+import Generators.*
 import org.scalacheck.Prop.*
 
 import scala.util.Try

diff --git a/core/src/test/scala/torch/TensorSuite.scala b/core/src/test/scala/torch/TensorSuite.scala
@@ -16,9 +16,6 @@
 
 package torch
 
-import org.scalacheck.Prop.*
-import Generators.given
-
 class TensorSuite extends TensorCheckSuite {
 
   test("tensor properties") {