fix: Fixed error messages

UFRN-URNAI · Oct 4, 2024 · 6edabcd · 6edabcd
1 parent cc97280
commit 6edabcd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/urnai/rewards/reward_base.py b/urnai/rewards/reward_base.py
@@ -16,7 +16,8 @@ def get(
         terminated: bool, 
         truncated: bool
     ) -> int: 
-        raise NotImplementedError(...)
+        raise NotImplementedError("Get method not implemented. You should implement " +
+                                  "it in your RewardBase subclass.")
 
     @abstractmethod
     def reset(self) -> None: ...
diff --git a/urnai/sc2/environments/stablebaselines3/custom_env.py b/urnai/sc2/environments/stablebaselines3/custom_env.py
@@ -53,7 +53,8 @@ def reset(
         return obs, info
 
     def render(self, mode: str) -> None:
-        raise NotImplementedError(...)
+        raise NotImplementedError("Render method not implemented. If necessary, you " +
+                                  "should implement it in your CustomEnv subclass.")
 
     def close(self) -> None:
         self._env.close()