diff --git a/cosmos_framework/data/vfm/action/datasets/base_dataset.py b/cosmos_framework/data/vfm/action/datasets/base_dataset.py
index 564d48e..a93e6ce 100644
--- a/cosmos_framework/data/vfm/action/datasets/base_dataset.py
+++ b/cosmos_framework/data/vfm/action/datasets/base_dataset.py
@@ -57,6 +57,7 @@ def __init__(
         self._pose_convention = pose_convention
         self._tolerance_s = float(tolerance_s)
         self._viewpoint = viewpoint
+        self._domain_name = domain_name
         self._domain_id = get_domain_id(domain_name)
         self._action_normalization = action_normalization
         self._norm_stats: dict[str, torch.Tensor] | None = None
@@ -97,6 +98,14 @@ def mode(self) -> str:
     def mode(self, value: str) -> None:
         self._mode = value
 
+    @property
+    def domain_name(self) -> str:
+        return self._domain_name
+
+    @property
+    def viewpoint(self) -> str:
+        return self._viewpoint
+
     @property
     def domain_id(self) -> int:
         return self._domain_id