gr-libs 0.1.5__py3-none-any.whl → 0.1.6.post1__py3-none-any.whl

CI/README.md

@@ -0,0 +1,12 @@
+## How to build a new docker image including new trained agents:
+1. Install docker
+2. Make sure you have a dataset.zip at your repo root
+3. Make sure you have a classic token in github: https://github.com/settings/tokens . If you don't, create one with package write, read and delete permissions and copy it somewhere safe.
+4. Authenticate to ghcr with docker by running:
+```sh
+echo ghp_REST_OF_TOKEN | docker login ghcr.io -u MatanShamir1 --password-stdin
+```
+3. docker build -t ghcr.io/<your-username>/gr_test_base:latest -f CI/Dockerfile .
+(the -f Dockerfile tells docker which Dockerfile to use and the '.' tells docker what's the build context, or where the dataset.zip should live)
+4. docker push ghcr.io/<your-username>/gr_test_base:latest
+docker push ghcr.io/MatanShamir1/gr_test_base:latest

CI/docker_build_context/Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+
+# Set workdir
+WORKDIR /app
+
+# Install unzip
+RUN apt-get update && apt-get install -y unzip && rm -rf /var/lib/apt/lists/*
+
+# Copy and unzip the dataset
+COPY dataset.zip .
+RUN unzip dataset.zip && rm dataset.zip
+RUN mv dataset_new dataset
+
+# Just start with bash by default
+CMD [ "bash" ]

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.6.post1'
+__version_tuple__ = version_tuple = (0, 1, 6)

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", "point_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):