fix issues

Author: Han Wang

deepmd/dpmodel/utils/type_embed.py

@@ -210,7 +210,6 @@ def change_type_map(
                 # Create random params with same dtype and device as first_layer_matrix
                 extend_type_params = np.random.default_rng().random(
                     [len(type_map), first_layer_matrix.shape[-1]],
-                    dtype=PRECISION_DICT[self.precision],
                 )
                 extend_type_params = xp.asarray(
                     extend_type_params,

source/tests/pt_expt/atomic_model/test_dp_atomic_model.py

@@ -142,6 +142,7 @@ def test_exportable(self) -> None:
         self.assertIn("energy", ret0)
 
         # Test torch.export
+        # Use strict=False for now to handle dynamic shapes
         exported = torch.export.export(
             md0,
             (coord, atype, nlist),