gr-libs 0.1.5__py3-none-any.whl → 0.1.7.post0__py3-none-any.whl

gr_libs/__init__.py

@@ -1,2 +1,6 @@
 from gr_libs.recognizer.graml.graml_recognizer import ExpertBasedGraml, GCGraml
-from gr_libs.recognizer.gr_as_rl.gr_as_rl_recognizer import Graql
+from gr_libs.recognizer.gr_as_rl.gr_as_rl_recognizer import Graql
+try:
+    from ._version import version as __version__
+except ImportError:
+    __version__ = "0.0.0"  # fallback if file isn't present

gr_libs/_version.py

@@ -0,0 +1,21 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+else:
+    VERSION_TUPLE = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+
+__version__ = version = '0.1.7.post0'
+__version_tuple__ = version_tuple = (0, 1, 7, 'post0')

gr_libs/environment/__init__.py

@@ -12,11 +12,11 @@ def is_extra_installed(package: str, extra: str) -> bool:
         return False  # The package is not installed
 
 # Check if `gr_libs[minigrid]` was installed
-for env in ["minigrid", "panda", "parking", "point_maze"]:
+for env in ["minigrid", "panda", "highway", "maze"]:
 	if is_extra_installed("gr_libs", f"gr_envs[{env}]"):
 		try:
 			importlib.import_module(f"gr_envs.{env}_scripts.envs")
 		except ImportError:
-			raise ImportError(f"gr_libs[{env}] was not installed, but gr_libs[{env}] requires it! if you messed with gr_libs installation, you can reinstall gr_libs.")
+			raise ImportError(f"gr_envs[{env}] was not installed, but gr_libs[{env}] requires it! if you messed with gr_envs installation, you can reinstall gr_libs.")
 	else:
 		warnings.warn(f"gr_libs[{env}] was not installed, skipping {env} imports.", RuntimeWarning)

gr_libs/environment/environment.py

@@ -105,7 +105,7 @@ class MinigridProperty(EnvProperty):
         env_id = problem_name.split("-DynamicGoal-")[0] + "-DynamicGoal-" + problem_name.split("-DynamicGoal-")[1]
         result = register(
             id=env_id,
-            entry_point="gr_libss.minigrid_scripts.envs:CustomColorEnv",
+            entry_point="gr_envs.minigrid_scripts.envs:CustomColorEnv",
             kwargs={"size": 13 if 'Simple' in problem_name else 9,
                     "num_crossings": 4 if 'Simple' in problem_name else 3,
                     "goal_pos": self.str_to_goal(problem_name),

gr_libs/metrics/metrics.py

@@ -5,7 +5,6 @@ import numpy as np
 
 from typing import Callable, Generator, List, Dict, Tuple, Any
 from math import log2
-from numpy.core.fromnumeric import mean
 from scipy.stats import wasserstein_distance
 from gymnasium.spaces.discrete import Discrete
 # import torch
@@ -43,7 +42,7 @@ def kl_divergence_norm_softmax(observations: List[Tuple[State, Any]], agent, act
         qp2_flatten_distribution_list: List[float] = agent.get_actions_probabilities(
             observation=(observation, agent_pos))
         distances.append(kl_divergence(qp1, qp2_flatten_distribution_list))
-    return mean(distances)
+    return np.mean(distances)
 
 
 def amplify(values, alpha=1.0):

gr_libs/ml/tabular/tabular_q_learner.py

@@ -351,7 +351,7 @@ class TabularQLearner(TabularRLAgent):
     def simplify_observation(self, observation):
         return [(obs['direction'], agent_pos_x, agent_pos_y, action) for ((obs, (agent_pos_x, agent_pos_y)), action) in observation] # list of tuples, each tuple the sample
         
-    def generate_observation(self, action_selection_method: MethodType, random_optimalism, save_fig = False, fig_path: str=None, env_prop=None):
+    def generate_observation(self, action_selection_method: MethodType, random_optimalism, save_fig=False, fig_path: str=None, env_prop=None):
         """
         Generate a single observation given a list of agents
 

gr_libs/ml/utils/storage.py

@@ -15,6 +15,13 @@ def get_storage_framework_dir(recognizer: str):
     return os.path.join(get_storage_dir(),recognizer)
 
 def get_storage_dir():
+    # Prefer local directory if it exists (e.g., in GitHub workspace)
+    if os.path.exists("dataset"):
+        return "dataset"
+    # Fall back to pre-mounted directory (e.g., in Docker container)
+    if os.path.exists("/preloaded_data"):
+        return "/preloaded_data"
+    # Default to "dataset" even if it doesn't exist (e.g., will be created)
     return "dataset"
 
 def _get_models_directory_name():