Fix tests

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

Hide file tree

Showing 5 changed files with 7 additions and 17 deletions.
diff --git a/core/src/main/scala/torch/hub.scala b/core/src/main/scala/torch/hub.scala
@@ -34,7 +34,6 @@ object hub:
     if !os.exists(cachedFile) then
       System.err.println(s"Downloading: $url to $cachedFile")
       Using.resource(URL(url).openStream()) { inputStream =>
-        Files.copy(inputStream, cachedFile.toNIO)
-        ()
+        val _ = Files.copy(inputStream, cachedFile.toNIO)
       }
     torch.pickleLoad(cachedFile.toNIO)
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") {