diff --git a/README.md b/README.md
index b7cf3d3..81f5f2c 100644
--- a/README.md
+++ b/README.md
@@ -1,9 +1,9 @@
-# gym-cache
+# gym_cache
OpenAI based Gym environments for training RL caching agent
install it:
- pip install gym-cache
+ pip install gym_cache
import it like this:
@@ -11,7 +11,7 @@ import it like this:
import gym
-gym.make('gym-cache:Cache-v0')
+gym.make('gym_cache:Cache-v0')
diff --git a/gym-cache/__init__.py b/gym-cache/__init__.py
index fc566a6..51aec57 100644
--- a/gym-cache/__init__.py
+++ b/gym-cache/__init__.py
@@ -6,7 +6,7 @@
# discrete action cache
register(
id='Cache-v0',
- entry_point='gym-cache.envs:CacheEnv',
+ entry_point='gym_cache.envs:CacheEnv',
kwargs={
'InputData': 'data/MWT2_processed',
'CacheSize': 100 * 1024 * 1024 * 1024
@@ -18,7 +18,7 @@
register(
id='Cache-large-v0',
- entry_point='gym-cache.envs:CacheEnv',
+ entry_point='gym_cache.envs:CacheEnv',
kwargs={
'InputData': 'data/MWT2_processed',
'CacheSize': 100 * 1024 * 1024 * 1024
@@ -31,7 +31,7 @@
# continuous action cache
register(
id='Cache-continuous-v0',
- entry_point='gym-cache.envs:CacheContinousEnv',
+ entry_point='gym_cache.envs:CacheContinousEnv',
kwargs={
'InputData': 'data/MWT2_processed',
'CacheSize': 100 * 1024 * 1024 * 1024
diff --git a/gym-cache/envs/__init__.py b/gym-cache/envs/__init__.py
index e6664ec..d33bb53 100644
--- a/gym-cache/envs/__init__.py
+++ b/gym-cache/envs/__init__.py
@@ -1,2 +1,2 @@
-from gym-cache.envs.cache_env import CacheEnv
-from gym-cache.envs.cache_continous import CacheContinousEnv
+from gym_cache.envs.cache_env import CacheEnv
+from gym_cache.envs.cache_continous import CacheContinousEnv
diff --git a/setup.py b/setup.py
index ce04e61..06e5810 100644
--- a/setup.py
+++ b/setup.py
@@ -4,8 +4,8 @@
long_description = fh.read()
setuptools.setup(
- name='gym-cache',
- version='1.0.0',
+ name='gym_cache',
+ version='1.0.1',
install_requires=[
'gym>=0.2.3',
'pandas>=0.24.2',
@@ -17,7 +17,7 @@
description="gym environment simulating file cache.",
long_description=long_description,
long_description_content_type="text/markdown",
- url="https://github.com/ivukotic/gym-cache",
+ url="https://github.com/ivukotic/gym_cache",
packages=setuptools.find_packages(),
classifiers=[
"Programming Language :: Python :: 3",
diff --git a/unit.py b/unit.py
index f98d595..d093eec 100644
--- a/unit.py
+++ b/unit.py
@@ -2,7 +2,7 @@
import gym
-env = gym.make('gym-cache:Cache-v0')
+env = gym.make('gym_cache:Cache-v0')
env.reset()
total_reward = 0