Use default_factory instead of an instance for this dataclass field's…

… default value. PiperOrigin-RevId: 547546481 Change-Id: I21ea27e026908bfd0390e5c772b3473fc46bb23d
google-deepmind · Jul 12, 2023 · fb87fd5 · fb87fd5
1 parent 36798a0
commit fb87fd5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/acme/agents/jax/mpo/config.py b/acme/agents/jax/mpo/config.py
@@ -32,8 +32,9 @@ class MPOConfig:
   discrete_policy: bool = False
 
   # Specification of the type of experience the learner will consume.
-  experience_type: mpo_types.ExperienceType = mpo_types.FromTransitions(
-      n_step=5)
+  experience_type: mpo_types.ExperienceType = dataclasses.field(
+      default_factory=lambda: mpo_types.FromTransitions(n_step=5)
+  )
   num_stacked_observations: int = 1
   # Optional data-augmentation transformation for observations.
   observation_transform: Optional[Callable[[types.NestedTensor],