fix test

OutisLi · OutisLi · commit 6846001b389b · 2026-02-21T19:56:44.000+08:00
diff --git a/source/tests/tf/test_lr.py b/source/tests/tf/test_lr.py
@@ -13,7 +13,6 @@
     LearningRateExp,
 )
 from deepmd.tf.env import (
-    GLOBAL_TF_FLOAT_PRECISION,
     tf,
 )
 from deepmd.tf.utils.learning_rate import (
@@ -49,7 +48,7 @@ def test_build_returns_tensor(self) -> None:
         lr_tensor = lr_schedule.build(global_step, num_steps=10000)
 
         self.assertIsInstance(lr_tensor, tf.Tensor)
-        self.assertEqual(lr_tensor.dtype, GLOBAL_TF_FLOAT_PRECISION)
+        self.assertEqual(lr_tensor.dtype, tf.float32)
 
     def test_default_type_exp(self) -> None:
         """Test that default type is 'exp' when not specified."""

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,6 @@`
`13`	`13`	`LearningRateExp,`
`14`	`14`	`)`
`15`	`15`	`from deepmd.tf.env import (`
`16`		`- GLOBAL_TF_FLOAT_PRECISION,`
`17`	`16`	`tf,`
`18`	`17`	`)`
`19`	`18`	`from deepmd.tf.utils.learning_rate import (`
`@@ -49,7 +48,7 @@ def test_build_returns_tensor(self) -> None:`
`49`	`48`	`lr_tensor = lr_schedule.build(global_step, num_steps=10000)`
`50`	`49`
`51`	`50`	`self.assertIsInstance(lr_tensor, tf.Tensor)`
`52`		`- self.assertEqual(lr_tensor.dtype, GLOBAL_TF_FLOAT_PRECISION)`
	`51`	`+ self.assertEqual(lr_tensor.dtype, tf.float32)`
`53`	`52`
`54`	`53`	`def test_default_type_exp(self) -> None:`
`55`	`54`	`"""Test that default type is 'exp' when not specified."""`