From 7752b5c2322bfe4610a6057821f2a422a4ce3993 Mon Sep 17 00:00:00 2001 From: Benjamin Meyer Date: Mon, 31 Aug 2026 08:42:41 +0200 Subject: [PATCH] Make jaxTypeName lazy as it accesses Python/JAX --- core/src/main/scala/dimwit/tensor/Tensor.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/src/main/scala/dimwit/tensor/Tensor.scala b/core/src/main/scala/dimwit/tensor/Tensor.scala index 027dba50..6ea683f9 100644 --- a/core/src/main/scala/dimwit/tensor/Tensor.scala +++ b/core/src/main/scala/dimwit/tensor/Tensor.scala @@ -72,7 +72,7 @@ class Tensor[T <: Tuple: Labels, V] private[dimwit] ( def extent[L: Label](axis: Axis[L])(using ev: AxisIndex[T, L]): AxisExtent[L] = shape.extent(axis) - private val jaxTypeName: String = py.Dynamic.global.`type`(jaxValue).`__name__`.as[String] + private lazy val jaxTypeName: String = py.Dynamic.global.`type`(jaxValue).`__name__`.as[String] lazy val isTracer: Boolean = val jaxCoreTracer = py.module("jax.core").Tracer py.Dynamic.global.isinstance(jaxValue, jaxCoreTracer).as[Boolean]