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