From 5c1fe2634ebfc4475ce93447a179a79eb4182bf7 Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 15:38:20 +0200
Subject: [PATCH 1/6] update pygame version

---
 .idea/misc.xml                                   |  2 +-
 .idea/server-simulator.iml                       |  3 ++-
 .../__init__.py                                  |  0
 src/{ => server_simulator}/envs/__init__.py      | 10 +++++-----
 .../envs/cluster_simulator}/__init__.py          |  0
 .../envs/cluster_simulator/actions.py            |  2 +-
 .../envs/cluster_simulator/base}/__init__.py     |  0
 .../base/extractors}/__init__.py                 |  0
 .../base/extractors/information.py               |  2 +-
 .../base/extractors/observation.py               |  6 +++---
 .../cluster_simulator/base/extractors/reward.py  |  4 ++--
 .../cluster_simulator/base/internal}/__init__.py |  0
 .../cluster_simulator/base/internal/cluster.py   |  6 +++---
 .../cluster_simulator/base/internal/dilation.py  |  0
 .../envs/cluster_simulator/base/internal/job.py  |  0
 .../cluster_simulator/base/internal/machine.py   |  0
 .../envs/cluster_simulator/base/renderer.py      |  4 ++--
 .../envs/cluster_simulator/basic.py              | 10 +++++-----
 .../envs/cluster_simulator/deep_rm/__init__.py   | 10 +++++-----
 .../envs/cluster_simulator/deep_rm/creator.py    | 10 +++++-----
 .../deep_rm/internal/__init__.py                 |  0
 .../deep_rm/internal/custom_type.py              |  0
 .../cluster_simulator/deep_rm/internal/jobs.py   |  4 ++--
 .../deep_rm/internal/machines.py                 |  4 ++--
 .../cluster_simulator/deep_rm/observation.py     |  6 +++---
 .../cluster_simulator/metric_based/__init__.py   | 10 +++++-----
 .../cluster_simulator/metric_based/creator.py    | 10 +++++-----
 .../metric_based/internal/__init__.py            |  0
 .../metric_based/internal/custom_type.py         |  0
 .../metric_based/internal/dilation.py            |  4 ++--
 .../metric_based/internal/jobs.py                |  4 ++--
 .../metric_based/internal/machines.py            |  4 ++--
 .../metric_based/observation.py                  | 10 +++++-----
 .../cluster_simulator/metric_based/renderer.py   | 10 +++++-----
 .../cluster_simulator/single_slot/__init__.py    |  6 +++---
 .../cluster_simulator/single_slot/creator.py     | 10 +++++-----
 .../single_slot/internal/__init__.py             |  0
 .../single_slot/internal/jobs.py                 |  2 +-
 .../single_slot/internal/machines.py             |  2 +-
 .../cluster_simulator/single_slot/observation.py |  6 +++---
 .../envs/cluster_simulator/utils/__init__.py     |  0
 .../cluster_simulator/utils/array_operations.py  |  0
 src/{ => server_simulator}/scheduler/__init__.py |  0
 .../scheduler/base_scheduler.py                  |  4 ++--
 .../first_come_first_served_scheduler.py         |  6 +++---
 .../scheduler/random_scheduler.py                |  6 +++---
 .../scheduler/round_robin_scheduler.py           |  6 +++---
 .../scheduler/shortest_job_first_scheduler.py    |  6 +++---
 src/server_simulator/wrappers/__init__.py        |  0
 .../wrappers/cluster_simulator/__init__.py       |  0
 .../cluster_simulator/dilation_wrapper.py        | 12 +++++-------
 .../wrappers/cluster_simulator/render_wrapper.py | 10 +++++-----
 tests/strategies/cluster_strategies/deeprm_st.py |  2 +-
 tests/strategies/cluster_strategies/metric_st.py |  2 +-
 tests/strategies/cluster_strategies/proto.py     |  6 ++----
 .../cluster_strategies/single_slot_st.py         |  2 +-
 .../metric_cluster_dilator_st.py                 | 12 ++++--------
 tests/strategies/dilation_strategies/proto.py    |  2 +-
 tests/strategies/env_strategies/basic_env_st.py  | 16 ++++++++--------
 tests/test_e2e/test_e2e_deeprm.py                |  3 ++-
 tests/test_e2e/test_e2e_matric_based.py          |  2 +-
 tests/test_e2e/test_e2e_single_slot.py           |  2 +-
 tests/test_envs/test_basic_env.py                | 12 ++++++------
 .../test_deep_rm/test_deeprm_cluster.py          |  9 ++++-----
 .../test_metric_based/test_metric_cluster.py     |  6 +++---
 .../test_metric_based/tests_metric_dilation.py   |  6 +++---
 .../test_single_slot/test_single_slot_cluster.py |  6 +++---
 .../test_single_slot/utils.py                    |  2 +-
 .../test_utils/test_array_operations.py          |  2 +-
 tests/test_wrappers/test_dilation_wrapper.py     | 16 +++++++---------
 70 files changed, 155 insertions(+), 164 deletions(-)
 rename src/{envs/cluster_simulator => server_simulator}/__init__.py (100%)
 rename src/{ => server_simulator}/envs/__init__.py (79%)
 rename src/{envs/cluster_simulator/base => server_simulator/envs/cluster_simulator}/__init__.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/actions.py (93%)
 rename src/{envs/cluster_simulator/base/extractors => server_simulator/envs/cluster_simulator/base}/__init__.py (100%)
 rename src/{envs/cluster_simulator/utils => server_simulator/envs/cluster_simulator/base/extractors}/__init__.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/extractors/information.py (87%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/extractors/observation.py (82%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/extractors/reward.py (82%)
 rename src/{wrappers/cluster_simulator => server_simulator/envs/cluster_simulator/base/internal}/__init__.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/internal/cluster.py (93%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/internal/dilation.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/internal/job.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/internal/machine.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/base/renderer.py (83%)
 rename src/{ => server_simulator}/envs/cluster_simulator/basic.py (84%)
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/__init__.py (92%)
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/creator.py (74%)
 create mode 100644 src/server_simulator/envs/cluster_simulator/deep_rm/internal/__init__.py
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/internal/custom_type.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/internal/jobs.py (93%)
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/internal/machines.py (91%)
 rename src/{ => server_simulator}/envs/cluster_simulator/deep_rm/observation.py (91%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/__init__.py (92%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/creator.py (76%)
 create mode 100644 src/server_simulator/envs/cluster_simulator/metric_based/internal/__init__.py
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/internal/custom_type.py (100%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/internal/dilation.py (94%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/internal/jobs.py (93%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/internal/machines.py (90%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/observation.py (85%)
 rename src/{ => server_simulator}/envs/cluster_simulator/metric_based/renderer.py (97%)
 rename src/{ => server_simulator}/envs/cluster_simulator/single_slot/__init__.py (91%)
 rename src/{ => server_simulator}/envs/cluster_simulator/single_slot/creator.py (74%)
 create mode 100644 src/server_simulator/envs/cluster_simulator/single_slot/internal/__init__.py
 rename src/{ => server_simulator}/envs/cluster_simulator/single_slot/internal/jobs.py (94%)
 rename src/{ => server_simulator}/envs/cluster_simulator/single_slot/internal/machines.py (94%)
 rename src/{ => server_simulator}/envs/cluster_simulator/single_slot/observation.py (90%)
 create mode 100644 src/server_simulator/envs/cluster_simulator/utils/__init__.py
 rename src/{ => server_simulator}/envs/cluster_simulator/utils/array_operations.py (100%)
 rename src/{ => server_simulator}/scheduler/__init__.py (100%)
 rename src/{ => server_simulator}/scheduler/base_scheduler.py (83%)
 rename src/{ => server_simulator}/scheduler/first_come_first_served_scheduler.py (78%)
 rename src/{ => server_simulator}/scheduler/random_scheduler.py (81%)
 rename src/{ => server_simulator}/scheduler/round_robin_scheduler.py (85%)
 rename src/{ => server_simulator}/scheduler/shortest_job_first_scheduler.py (83%)
 create mode 100644 src/server_simulator/wrappers/__init__.py
 create mode 100644 src/server_simulator/wrappers/cluster_simulator/__init__.py
 rename src/{ => server_simulator}/wrappers/cluster_simulator/dilation_wrapper.py (92%)
 rename src/{ => server_simulator}/wrappers/cluster_simulator/render_wrapper.py (84%)

diff --git a/.idea/misc.xml b/.idea/misc.xml
index 61d7a0c..b6ce6a8 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,5 +3,5 @@
   <component name="Black">
     <option name="sdkName" value="Python 3.10 (server-simulator)" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (server-simulator)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.11 virtualenv at ~/Documents/Thesis/server-simulator/.venv" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/.idea/server-simulator.iml b/.idea/server-simulator.iml
index dc3d6fb..4166a26 100644
--- a/.idea/server-simulator.iml
+++ b/.idea/server-simulator.iml
@@ -3,8 +3,9 @@
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
       <excludeFolder url="file://$MODULE_DIR$/.venv" />
+      <excludeFolder url="file://$MODULE_DIR$/.venv1" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.10 (server-simulator)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.11 virtualenv at ~/Documents/Thesis/server-simulator/.venv" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>
\ No newline at end of file
diff --git a/src/envs/cluster_simulator/__init__.py b/src/server_simulator/__init__.py
similarity index 100%
rename from src/envs/cluster_simulator/__init__.py
rename to src/server_simulator/__init__.py
diff --git a/src/envs/__init__.py b/src/server_simulator/envs/__init__.py
similarity index 79%
rename from src/envs/__init__.py
rename to src/server_simulator/envs/__init__.py
index 7256ff9..537a7f7 100644
--- a/src/envs/__init__.py
+++ b/src/server_simulator/envs/__init__.py
@@ -1,18 +1,18 @@
-from src.envs.cluster_simulator.base.extractors.reward import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
     DifferentInPendingJobsRewardCaculator,
 )
-from src.envs.cluster_simulator.basic import BasicClusterEnv as BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv as BasicClusterEnv
 from gymnasium import register
 
-from src.envs.cluster_simulator.deep_rm.creator import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.creator import (
     DeepRMEnvCreator,
     DeepRMCreatorParameters,
 )
-from src.envs.cluster_simulator.metric_based.creator import (
+from src.server_simulator.envs.cluster_simulator.metric_based.creator import (
     MetricBasedEnvCreator,
     MetricBasedCreatorParameters,
 )
-from src.envs.cluster_simulator.single_slot.creator import (
+from src.server_simulator.envs.cluster_simulator.single_slot.creator import (
     SingleSlotEnvCreator,
     SingleSlotCreatorParameters,
 )
diff --git a/src/envs/cluster_simulator/base/__init__.py b/src/server_simulator/envs/cluster_simulator/__init__.py
similarity index 100%
rename from src/envs/cluster_simulator/base/__init__.py
rename to src/server_simulator/envs/cluster_simulator/__init__.py
diff --git a/src/envs/cluster_simulator/actions.py b/src/server_simulator/envs/cluster_simulator/actions.py
similarity index 93%
rename from src/envs/cluster_simulator/actions.py
rename to src/server_simulator/envs/cluster_simulator/actions.py
index 148b269..f67eb99 100644
--- a/src/envs/cluster_simulator/actions.py
+++ b/src/server_simulator/envs/cluster_simulator/actions.py
@@ -1,7 +1,7 @@
 from typing import NamedTuple, Tuple, TypeVar
 import gymnasium as gym
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterAction, ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterAction, ClusterABC
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
 
diff --git a/src/envs/cluster_simulator/base/extractors/__init__.py b/src/server_simulator/envs/cluster_simulator/base/__init__.py
similarity index 100%
rename from src/envs/cluster_simulator/base/extractors/__init__.py
rename to src/server_simulator/envs/cluster_simulator/base/__init__.py
diff --git a/src/envs/cluster_simulator/utils/__init__.py b/src/server_simulator/envs/cluster_simulator/base/extractors/__init__.py
similarity index 100%
rename from src/envs/cluster_simulator/utils/__init__.py
rename to src/server_simulator/envs/cluster_simulator/base/extractors/__init__.py
diff --git a/src/envs/cluster_simulator/base/extractors/information.py b/src/server_simulator/envs/cluster_simulator/base/extractors/information.py
similarity index 87%
rename from src/envs/cluster_simulator/base/extractors/information.py
rename to src/server_simulator/envs/cluster_simulator/base/extractors/information.py
index b340c96..95f9ccc 100644
--- a/src/envs/cluster_simulator/base/extractors/information.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/information.py
@@ -2,7 +2,7 @@
 from typing import TypeVar, TypedDict, Generic
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.extractors.observation import ClusterObservation
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import ClusterObservation
 
 
 class ClusterBaseInformation(TypedDict):
diff --git a/src/envs/cluster_simulator/base/extractors/observation.py b/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
similarity index 82%
rename from src/envs/cluster_simulator/base/extractors/observation.py
rename to src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
index b78785f..e49171b 100644
--- a/src/envs/cluster_simulator/base/extractors/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
@@ -3,9 +3,9 @@
 import gymnasium as gym
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
-from src.envs.cluster_simulator.base.internal.job import JobCollectionConvertor
-from src.envs.cluster_simulator.base.internal.machine import MachinesCollectionConvertor
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollectionConvertor
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachinesCollectionConvertor
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
 MachinesRepresentation = TypeVar("MachinesRepresentation")
diff --git a/src/envs/cluster_simulator/base/extractors/reward.py b/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
similarity index 82%
rename from src/envs/cluster_simulator/base/extractors/reward.py
rename to src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
index 9b5240b..596f177 100644
--- a/src/envs/cluster_simulator/base/extractors/reward.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
@@ -1,8 +1,8 @@
 from typing import Generic
 import abc
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.base.extractors.information import ClusterInformation
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
 
 
 class RewardCaculator(Generic[ClusterInformation]):
diff --git a/src/wrappers/cluster_simulator/__init__.py b/src/server_simulator/envs/cluster_simulator/base/internal/__init__.py
similarity index 100%
rename from src/wrappers/cluster_simulator/__init__.py
rename to src/server_simulator/envs/cluster_simulator/base/internal/__init__.py
diff --git a/src/envs/cluster_simulator/base/internal/cluster.py b/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
similarity index 93%
rename from src/envs/cluster_simulator/base/internal/cluster.py
rename to src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
index a70e1f6..060358c 100644
--- a/src/envs/cluster_simulator/base/internal/cluster.py
+++ b/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
@@ -3,9 +3,9 @@
 
 from rust_enum import enum, Case
 
-from src.envs.cluster_simulator.base.internal.job import Job, JobCollection
-from src.envs.cluster_simulator.base.internal.job import Status as JobStatus
-from src.envs.cluster_simulator.base.internal.machine import Machine, MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Job, JobCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status as JobStatus
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import Machine, MachineCollection
 import logging
 
 T = tp.TypeVar("T")
diff --git a/src/envs/cluster_simulator/base/internal/dilation.py b/src/server_simulator/envs/cluster_simulator/base/internal/dilation.py
similarity index 100%
rename from src/envs/cluster_simulator/base/internal/dilation.py
rename to src/server_simulator/envs/cluster_simulator/base/internal/dilation.py
diff --git a/src/envs/cluster_simulator/base/internal/job.py b/src/server_simulator/envs/cluster_simulator/base/internal/job.py
similarity index 100%
rename from src/envs/cluster_simulator/base/internal/job.py
rename to src/server_simulator/envs/cluster_simulator/base/internal/job.py
diff --git a/src/envs/cluster_simulator/base/internal/machine.py b/src/server_simulator/envs/cluster_simulator/base/internal/machine.py
similarity index 100%
rename from src/envs/cluster_simulator/base/internal/machine.py
rename to src/server_simulator/envs/cluster_simulator/base/internal/machine.py
diff --git a/src/envs/cluster_simulator/base/renderer.py b/src/server_simulator/envs/cluster_simulator/base/renderer.py
similarity index 83%
rename from src/envs/cluster_simulator/base/renderer.py
rename to src/server_simulator/envs/cluster_simulator/base/renderer.py
index a2bfd31..4171e84 100644
--- a/src/envs/cluster_simulator/base/renderer.py
+++ b/src/server_simulator/envs/cluster_simulator/base/renderer.py
@@ -1,10 +1,10 @@
 from types import TracebackType
 from typing import TypeVar, Generic
 from abc import abstractmethod
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     ClusterBaseInformation,
 )
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
     BaseClusterObservation,
 )
 from typing_extensions import Self
diff --git a/src/envs/cluster_simulator/basic.py b/src/server_simulator/envs/cluster_simulator/basic.py
similarity index 84%
rename from src/envs/cluster_simulator/basic.py
rename to src/server_simulator/envs/cluster_simulator/basic.py
index 861573a..23516e1 100644
--- a/src/envs/cluster_simulator/basic.py
+++ b/src/server_simulator/envs/cluster_simulator/basic.py
@@ -2,17 +2,17 @@
 import typing as tp
 import numpy as np
 
-from src.envs.cluster_simulator.actions import EnvironmentAction, ActionConvertor
-from src.envs.cluster_simulator.base.extractors.reward import RewardCaculator
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.actions import EnvironmentAction, ActionConvertor
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     ClusterInformation,
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
     ClusterObservation,
     BaseObservationCreatorProtocol,
 )
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
 
 InputActType = np.int64
 T = tp.TypeVar("T", bound=type)
diff --git a/src/envs/cluster_simulator/deep_rm/__init__.py b/src/server_simulator/envs/cluster_simulator/deep_rm/__init__.py
similarity index 92%
rename from src/envs/cluster_simulator/deep_rm/__init__.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/__init__.py
index c416cff..d4b2aed 100644
--- a/src/envs/cluster_simulator/deep_rm/__init__.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/__init__.py
@@ -2,26 +2,26 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 
 
-from src.envs.cluster_simulator.deep_rm.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.internal.custom_type import (
     _JOBS_TYPE as _JOBS_TYPE,
     _MACHINE_TYPE as _MACHINE_TYPE,
     _DTYPE as _DTYPE,
 )
-from src.envs.cluster_simulator.deep_rm.internal.jobs import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.internal.jobs import (
     DeepRMJobs,
     DeepRMJobSlot,
     DeepRMJobsConvertor as DeepRMJobsConvertor,
 )
-from src.envs.cluster_simulator.deep_rm.internal.machines import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.internal.machines import (
     DeepRMMachine,
     DeepRMMachines,
     DeepRMMachinesConvertor as DeepRMMachinesConvertor,
 )
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
 
 
 class DeepRMCluster(ClusterABC[DeepRMMachines, DeepRMJobs]):
diff --git a/src/envs/cluster_simulator/deep_rm/creator.py b/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
similarity index 74%
rename from src/envs/cluster_simulator/deep_rm/creator.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
index b19b8bb..6b673dd 100644
--- a/src/envs/cluster_simulator/deep_rm/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
@@ -1,14 +1,14 @@
 from gymnasium.envs.registration import EnvCreator
 
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.base.extractors.reward import RewardCaculator
-from src.envs.cluster_simulator.basic import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
-from src.envs.cluster_simulator.deep_rm import DeepRMCluster, DeepRMCreators
-from src.envs.cluster_simulator.deep_rm.observation import DeepRMObservationCreator
+from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMCluster, DeepRMCreators
+from src.server_simulator.envs.cluster_simulator.deep_rm.observation import DeepRMObservationCreator
 
 __all__ = ["DeepRMCreatorParameters", "DeepRMEnvCreator"]
 
diff --git a/src/server_simulator/envs/cluster_simulator/deep_rm/internal/__init__.py b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/envs/cluster_simulator/deep_rm/internal/custom_type.py b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/custom_type.py
similarity index 100%
rename from src/envs/cluster_simulator/deep_rm/internal/custom_type.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/internal/custom_type.py
diff --git a/src/envs/cluster_simulator/deep_rm/internal/jobs.py b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/jobs.py
similarity index 93%
rename from src/envs/cluster_simulator/deep_rm/internal/jobs.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/internal/jobs.py
index c022754..ab2c05c 100644
--- a/src/envs/cluster_simulator/deep_rm/internal/jobs.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/jobs.py
@@ -4,13 +4,13 @@
 import numpy.typing as npt
 from typing import TypeAlias
 from typing_extensions import Unpack
-from src.envs.cluster_simulator.base.internal.job import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
     Job,
     JobCollection,
     Status,
     JobCollectionConvertor,
 )
-from src.envs.cluster_simulator.deep_rm.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.internal.custom_type import (
     _JOB_TYPE,
     _JOBS_TYPE,
 )
diff --git a/src/envs/cluster_simulator/deep_rm/internal/machines.py b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/machines.py
similarity index 91%
rename from src/envs/cluster_simulator/deep_rm/internal/machines.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/internal/machines.py
index b60657a..f888fcc 100644
--- a/src/envs/cluster_simulator/deep_rm/internal/machines.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/internal/machines.py
@@ -1,12 +1,12 @@
 import typing as tp
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.machine import (
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
     Machine,
     MachineCollection,
     MachinesCollectionConvertor,
 )
-from src.envs.cluster_simulator.deep_rm.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.internal.custom_type import (
     _MACHINE_TYPE,
     _MACHINES_TYPE,
 )
diff --git a/src/envs/cluster_simulator/deep_rm/observation.py b/src/server_simulator/envs/cluster_simulator/deep_rm/observation.py
similarity index 91%
rename from src/envs/cluster_simulator/deep_rm/observation.py
rename to src/server_simulator/envs/cluster_simulator/deep_rm/observation.py
index 1253bf2..e813135 100644
--- a/src/envs/cluster_simulator/deep_rm/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/observation.py
@@ -4,11 +4,11 @@
 import numpy as np
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
     BaseObservationCreatorProtocol,
 )
-from src.envs.cluster_simulator.deep_rm import (
+from src.server_simulator.envs.cluster_simulator.deep_rm import (
     DeepRMCluster,
     DeepRMMachinesConvertor,
     DeepRMJobsConvertor,
diff --git a/src/envs/cluster_simulator/metric_based/__init__.py b/src/server_simulator/envs/cluster_simulator/metric_based/__init__.py
similarity index 92%
rename from src/envs/cluster_simulator/metric_based/__init__.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/__init__.py
index 474eed5..8360640 100644
--- a/src/envs/cluster_simulator/metric_based/__init__.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/__init__.py
@@ -2,24 +2,24 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.metric_based.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.custom_type import (
     _JOBS_TYPE as _JOBS_TYPE,
     _MACHINE_TYPE as _MACHINE_TYPE,
     _DTYPE as _DTYPE,
 )
-from src.envs.cluster_simulator.metric_based.internal.jobs import (
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.jobs import (
     MetricJobSlot,
     MetricJobs,
     MetricJobsConvertor as MetricJobsConvertor,
 )
-from src.envs.cluster_simulator.metric_based.internal.machines import (
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.machines import (
     MetricMachine,
     MetricMachines,
     MetricMachinesConvertor as MetricMachinesConvertor,
 )
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
 
 
 class MetricCluster(ClusterABC[MetricMachines, MetricJobs]):
diff --git a/src/envs/cluster_simulator/metric_based/creator.py b/src/server_simulator/envs/cluster_simulator/metric_based/creator.py
similarity index 76%
rename from src/envs/cluster_simulator/metric_based/creator.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/creator.py
index 6f075ac..69a310f 100644
--- a/src/envs/cluster_simulator/metric_based/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/creator.py
@@ -1,15 +1,15 @@
 from gymnasium.envs.registration import EnvCreator
 
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.base.extractors.reward import RewardCaculator
-from src.envs.cluster_simulator.basic import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
 
-from src.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
-from src.envs.cluster_simulator.metric_based.observation import (
+from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
+from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
 
diff --git a/src/server_simulator/envs/cluster_simulator/metric_based/internal/__init__.py b/src/server_simulator/envs/cluster_simulator/metric_based/internal/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/envs/cluster_simulator/metric_based/internal/custom_type.py b/src/server_simulator/envs/cluster_simulator/metric_based/internal/custom_type.py
similarity index 100%
rename from src/envs/cluster_simulator/metric_based/internal/custom_type.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/internal/custom_type.py
diff --git a/src/envs/cluster_simulator/metric_based/internal/dilation.py b/src/server_simulator/envs/cluster_simulator/metric_based/internal/dilation.py
similarity index 94%
rename from src/envs/cluster_simulator/metric_based/internal/dilation.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/internal/dilation.py
index 4a32cd8..1bdcaf8 100644
--- a/src/envs/cluster_simulator/metric_based/internal/dilation.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/internal/dilation.py
@@ -2,11 +2,11 @@
 import numpy.typing as npt
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.dilation import (
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
     AbstractDilation,
     SelectCellAction,
 )
-from src.envs.cluster_simulator.utils.array_operations import (
+from src.server_simulator.envs.cluster_simulator.utils.array_operations import (
     hierarchical_pooling,
     get_window_from_cell,
 )
diff --git a/src/envs/cluster_simulator/metric_based/internal/jobs.py b/src/server_simulator/envs/cluster_simulator/metric_based/internal/jobs.py
similarity index 93%
rename from src/envs/cluster_simulator/metric_based/internal/jobs.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/internal/jobs.py
index 5387f20..88ec59b 100644
--- a/src/envs/cluster_simulator/metric_based/internal/jobs.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/internal/jobs.py
@@ -4,13 +4,13 @@
 import numpy as np
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.job import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
     Job,
     Status,
     JobCollection,
     JobCollectionConvertor,
 )
-from src.envs.cluster_simulator.metric_based.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.custom_type import (
     _JOBS_TYPE,
     _JOB_TYPE,
 )
diff --git a/src/envs/cluster_simulator/metric_based/internal/machines.py b/src/server_simulator/envs/cluster_simulator/metric_based/internal/machines.py
similarity index 90%
rename from src/envs/cluster_simulator/metric_based/internal/machines.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/internal/machines.py
index f6fa713..fe6e542 100644
--- a/src/envs/cluster_simulator/metric_based/internal/machines.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/internal/machines.py
@@ -3,12 +3,12 @@
 from typing_extensions import Unpack
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.machine import (
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
     Machine,
     MachineCollection,
     MachinesCollectionConvertor,
 )
-from src.envs.cluster_simulator.metric_based.internal.custom_type import (
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.custom_type import (
     _MACHINE_TYPE,
     _MACHINES_TYPE,
 )
diff --git a/src/envs/cluster_simulator/metric_based/observation.py b/src/server_simulator/envs/cluster_simulator/metric_based/observation.py
similarity index 85%
rename from src/envs/cluster_simulator/metric_based/observation.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/observation.py
index a734310..75c39d6 100644
--- a/src/envs/cluster_simulator/metric_based/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/observation.py
@@ -3,15 +3,15 @@
 import gymnasium as gym
 import numpy.typing as npt
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
     BaseObservationCreatorProtocol,
 )
 import numpy as np
 
-from src.envs.cluster_simulator.metric_based import MetricCluster
-from src.envs.cluster_simulator.metric_based.internal.jobs import MetricJobsConvertor
-from src.envs.cluster_simulator.metric_based.internal.machines import (
+from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.jobs import MetricJobsConvertor
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.machines import (
     MetricMachinesConvertor,
 )
 
diff --git a/src/envs/cluster_simulator/metric_based/renderer.py b/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
similarity index 97%
rename from src/envs/cluster_simulator/metric_based/renderer.py
rename to src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
index dd50b55..07f084a 100644
--- a/src/envs/cluster_simulator/metric_based/renderer.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
@@ -2,19 +2,19 @@
 
 from pygame import Color
 
-from src.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
-from src.envs.cluster_simulator.metric_based.observation import (
+from src.server_simulator.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
+from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservation,
 )
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     ClusterBaseInformation,
 )
-from src.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 import numpy.typing as npt
 import pygame
 import numpy as np
 
-from src.wrappers.cluster_simulator.render_wrapper import RenderMode
+from src.server_simulator.wrappers.cluster_simulator.render_wrapper import RenderMode
 
 
 class ClusterMetricRenderer(
diff --git a/src/envs/cluster_simulator/single_slot/__init__.py b/src/server_simulator/envs/cluster_simulator/single_slot/__init__.py
similarity index 91%
rename from src/envs/cluster_simulator/single_slot/__init__.py
rename to src/server_simulator/envs/cluster_simulator/single_slot/__init__.py
index 295bd95..30b215a 100644
--- a/src/envs/cluster_simulator/single_slot/__init__.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/__init__.py
@@ -2,14 +2,14 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
-from src.envs.cluster_simulator.single_slot.internal.jobs import (
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
+from src.server_simulator.envs.cluster_simulator.single_slot.internal.jobs import (
     SingleSlotJobs,
     Status,
     SingleSlotJob,
     SingleSlotJobsConvertor as SingleSlotJobsConvertor,
 )
-from src.envs.cluster_simulator.single_slot.internal.machines import (
+from src.server_simulator.envs.cluster_simulator.single_slot.internal.machines import (
     SingleSlotMachines,
     SingleSlotMachine,
     SingleSlotMachinesConvertor as SingleSlotMachinesConvertor,
diff --git a/src/envs/cluster_simulator/single_slot/creator.py b/src/server_simulator/envs/cluster_simulator/single_slot/creator.py
similarity index 74%
rename from src/envs/cluster_simulator/single_slot/creator.py
rename to src/server_simulator/envs/cluster_simulator/single_slot/creator.py
index 63e66ed..acdc963 100644
--- a/src/envs/cluster_simulator/single_slot/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/creator.py
@@ -1,18 +1,18 @@
 from gymnasium.envs.registration import EnvCreator
 
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.base.extractors.reward import RewardCaculator
-from src.envs.cluster_simulator.basic import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
 
-from src.envs.cluster_simulator.single_slot import (
+from src.server_simulator.envs.cluster_simulator.single_slot import (
     SingleSlotCluster,
     SingleSlotClusterCreators,
 )
-from src.envs.cluster_simulator.single_slot.observation import (
+from src.server_simulator.envs.cluster_simulator.single_slot.observation import (
     SingleSlotObservationCreator,
 )
 
diff --git a/src/server_simulator/envs/cluster_simulator/single_slot/internal/__init__.py b/src/server_simulator/envs/cluster_simulator/single_slot/internal/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/envs/cluster_simulator/single_slot/internal/jobs.py b/src/server_simulator/envs/cluster_simulator/single_slot/internal/jobs.py
similarity index 94%
rename from src/envs/cluster_simulator/single_slot/internal/jobs.py
rename to src/server_simulator/envs/cluster_simulator/single_slot/internal/jobs.py
index 2467dfd..a7b8671 100644
--- a/src/envs/cluster_simulator/single_slot/internal/jobs.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/internal/jobs.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
     Job,
     JobCollection,
     Status,
diff --git a/src/envs/cluster_simulator/single_slot/internal/machines.py b/src/server_simulator/envs/cluster_simulator/single_slot/internal/machines.py
similarity index 94%
rename from src/envs/cluster_simulator/single_slot/internal/machines.py
rename to src/server_simulator/envs/cluster_simulator/single_slot/internal/machines.py
index ef912eb..caab1e9 100644
--- a/src/envs/cluster_simulator/single_slot/internal/machines.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/internal/machines.py
@@ -2,7 +2,7 @@
 from typing import TypeAlias
 from typing_extensions import Unpack
 import numpy as np
-from src.envs.cluster_simulator.base.internal.machine import (
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
     Machine,
     MachineCollection,
     MachinesCollectionConvertor,
diff --git a/src/envs/cluster_simulator/single_slot/observation.py b/src/server_simulator/envs/cluster_simulator/single_slot/observation.py
similarity index 90%
rename from src/envs/cluster_simulator/single_slot/observation.py
rename to src/server_simulator/envs/cluster_simulator/single_slot/observation.py
index 193471f..7b8d290 100644
--- a/src/envs/cluster_simulator/single_slot/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/observation.py
@@ -2,11 +2,11 @@
 
 import gymnasium as gym
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
     BaseObservationCreatorProtocol,
 )
-from src.envs.cluster_simulator.single_slot import (
+from src.server_simulator.envs.cluster_simulator.single_slot import (
     SingleSlotCluster,
     SingleSlotMachinesConvertor,
     SingleSlotJobsConvertor,
diff --git a/src/server_simulator/envs/cluster_simulator/utils/__init__.py b/src/server_simulator/envs/cluster_simulator/utils/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/envs/cluster_simulator/utils/array_operations.py b/src/server_simulator/envs/cluster_simulator/utils/array_operations.py
similarity index 100%
rename from src/envs/cluster_simulator/utils/array_operations.py
rename to src/server_simulator/envs/cluster_simulator/utils/array_operations.py
diff --git a/src/scheduler/__init__.py b/src/server_simulator/scheduler/__init__.py
similarity index 100%
rename from src/scheduler/__init__.py
rename to src/server_simulator/scheduler/__init__.py
diff --git a/src/scheduler/base_scheduler.py b/src/server_simulator/scheduler/base_scheduler.py
similarity index 83%
rename from src/scheduler/base_scheduler.py
rename to src/server_simulator/scheduler/base_scheduler.py
index 8e8e53e..f191502 100644
--- a/src/scheduler/base_scheduler.py
+++ b/src/server_simulator/scheduler/base_scheduler.py
@@ -2,8 +2,8 @@
 import typing as tp
 import abc
 
-from src.envs.cluster_simulator.base.internal.job import JobCollection, Job, Status
-from src.envs.cluster_simulator.base.internal.machine import MachineCollection, Machine
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection, Job, Status
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection, Machine
 
 T = tp.TypeVar("T")
 MachineT = tp.TypeVar("MachineT", bound=Machine)
diff --git a/src/scheduler/first_come_first_served_scheduler.py b/src/server_simulator/scheduler/first_come_first_served_scheduler.py
similarity index 78%
rename from src/scheduler/first_come_first_served_scheduler.py
rename to src/server_simulator/scheduler/first_come_first_served_scheduler.py
index 3e14da9..1847bd0 100644
--- a/src/scheduler/first_come_first_served_scheduler.py
+++ b/src/server_simulator/scheduler/first_come_first_served_scheduler.py
@@ -1,8 +1,8 @@
 import typing as tp
 
-from src.envs.cluster_simulator.base.internal.job import JobCollection
-from src.envs.cluster_simulator.base.internal.machine import MachineCollection
-from src.scheduler.base_scheduler import ABCScheduler
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
 
diff --git a/src/scheduler/random_scheduler.py b/src/server_simulator/scheduler/random_scheduler.py
similarity index 81%
rename from src/scheduler/random_scheduler.py
rename to src/server_simulator/scheduler/random_scheduler.py
index f5d434f..f48cac0 100644
--- a/src/scheduler/random_scheduler.py
+++ b/src/server_simulator/scheduler/random_scheduler.py
@@ -2,9 +2,9 @@
 import random
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import JobCollection
-from src.envs.cluster_simulator.base.internal.machine import MachineCollection
-from src.scheduler.base_scheduler import ABCScheduler, T
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.scheduler.base_scheduler import ABCScheduler, T
 
 
 class RandomScheduler(ABCScheduler[T]):
diff --git a/src/scheduler/round_robin_scheduler.py b/src/server_simulator/scheduler/round_robin_scheduler.py
similarity index 85%
rename from src/scheduler/round_robin_scheduler.py
rename to src/server_simulator/scheduler/round_robin_scheduler.py
index 244f16d..330f0e8 100644
--- a/src/scheduler/round_robin_scheduler.py
+++ b/src/server_simulator/scheduler/round_robin_scheduler.py
@@ -1,8 +1,8 @@
 import typing as tp
 
-from src.envs.cluster_simulator.base.internal.job import JobCollection
-from src.envs.cluster_simulator.base.internal.machine import MachineCollection
-from src.scheduler.base_scheduler import ABCScheduler
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
 
diff --git a/src/scheduler/shortest_job_first_scheduler.py b/src/server_simulator/scheduler/shortest_job_first_scheduler.py
similarity index 83%
rename from src/scheduler/shortest_job_first_scheduler.py
rename to src/server_simulator/scheduler/shortest_job_first_scheduler.py
index dfe23a4..580e1bb 100644
--- a/src/scheduler/shortest_job_first_scheduler.py
+++ b/src/server_simulator/scheduler/shortest_job_first_scheduler.py
@@ -1,8 +1,8 @@
 import typing as tp
 
-from src.envs.cluster_simulator.base.internal.job import JobCollection
-from src.envs.cluster_simulator.base.internal.machine import MachineCollection
-from src.scheduler.base_scheduler import ABCScheduler
+from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
 
diff --git a/src/server_simulator/wrappers/__init__.py b/src/server_simulator/wrappers/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/server_simulator/wrappers/cluster_simulator/__init__.py b/src/server_simulator/wrappers/cluster_simulator/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/wrappers/cluster_simulator/dilation_wrapper.py b/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
similarity index 92%
rename from src/wrappers/cluster_simulator/dilation_wrapper.py
rename to src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
index 3884ea2..d0a34d6 100644
--- a/src/wrappers/cluster_simulator/dilation_wrapper.py
+++ b/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
@@ -6,18 +6,16 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.dilation import (
+from src.server_simulator.envs import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.actions import EnvironmentAction, DilationEnvironmentAction
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import BaseClusterObservation
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
     AbstractDilation,
     DilationAction,
     DilationState,
     AbstractDilationParams,
 )
-from src.envs.cluster_simulator.actions import DilationEnvironmentAction
-from src.envs.cluster_simulator.basic import BasicClusterEnv, EnvironmentAction
-from src.envs.cluster_simulator.base.extractors.information import ClusterInformation
-from src.envs.cluster_simulator.base.extractors.observation import (
-    BaseClusterObservation,
-)
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
 
 EnvironmentObservation = TypeVar("EnvironmentObservation", bound=BaseClusterObservation)
 WrapperObservation = TypeVar("WrapperObservation", bound=BaseClusterObservation)
diff --git a/src/wrappers/cluster_simulator/render_wrapper.py b/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
similarity index 84%
rename from src/wrappers/cluster_simulator/render_wrapper.py
rename to src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
index 9cacd72..298a242 100644
--- a/src/wrappers/cluster_simulator/render_wrapper.py
+++ b/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
@@ -3,15 +3,15 @@
 import gymnasium as gym
 from gymnasium.core import WrapperActType, WrapperObsType, RenderFrame
 
-from src.envs import BasicClusterEnv
-from src.envs.cluster_simulator.actions import EnvironmentAction
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator import EnvironmentAction
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     ClusterBaseInformation,
 )
-from src.envs.cluster_simulator.base.extractors.observation import (
+from src.server_simulator.envs.cluster_simulator import (
     BaseClusterObservation,
 )
-from src.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
+from src.server_simulator.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
 
 EnvironmentObservation = TypeVar("EnvironmentObservation", bound=BaseClusterObservation)
 WrapperObservation = TypeVar("WrapperObservation", bound=BaseClusterObservation)
diff --git a/tests/strategies/cluster_strategies/deeprm_st.py b/tests/strategies/cluster_strategies/deeprm_st.py
index b94ad14..38b65d6 100644
--- a/tests/strategies/cluster_strategies/deeprm_st.py
+++ b/tests/strategies/cluster_strategies/deeprm_st.py
@@ -1,7 +1,7 @@
 from hypothesis import strategies as st
 from hypothesis.strategies import SearchStrategy
 
-from src.envs.cluster_simulator.deep_rm import DeepRMCreators, DeepRMCluster
+from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMCluster, DeepRMCreators
 from tests.strategies.cluster_strategies.proto import ClusterStrategies
 
 
diff --git a/tests/strategies/cluster_strategies/metric_st.py b/tests/strategies/cluster_strategies/metric_st.py
index 2d5eb32..cb82f0a 100644
--- a/tests/strategies/cluster_strategies/metric_st.py
+++ b/tests/strategies/cluster_strategies/metric_st.py
@@ -1,7 +1,7 @@
 from hypothesis import strategies as st
 from hypothesis.strategies import SearchStrategy
 
-from src.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
+from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
 from tests.strategies.cluster_strategies.proto import ClusterStrategies
 
 
diff --git a/tests/strategies/cluster_strategies/proto.py b/tests/strategies/cluster_strategies/proto.py
index e32aa3a..e5395f9 100644
--- a/tests/strategies/cluster_strategies/proto.py
+++ b/tests/strategies/cluster_strategies/proto.py
@@ -3,10 +3,8 @@
 
 from hypothesis.strategies import SearchStrategy
 
-from src.envs.cluster_simulator.base.internal.cluster import ClusterABC
-from src.envs.cluster_simulator.base.extractors.observation import (
-    BaseObservationCreatorProtocol,
-)
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import BaseObservationCreatorProtocol
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
 Creator = TypeVar("Creator", bound=BaseObservationCreatorProtocol)
diff --git a/tests/strategies/cluster_strategies/single_slot_st.py b/tests/strategies/cluster_strategies/single_slot_st.py
index 93e62ce..ff305bb 100644
--- a/tests/strategies/cluster_strategies/single_slot_st.py
+++ b/tests/strategies/cluster_strategies/single_slot_st.py
@@ -2,7 +2,7 @@
 from hypothesis import strategies as st
 
 from tests.strategies.cluster_strategies.proto import ClusterStrategies
-from src.envs.cluster_simulator.single_slot import (
+from src.server_simulator.envs.cluster_simulator.single_slot import (
     SingleSlotCluster,
     SingleSlotClusterCreators,
 )
diff --git a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
index dc81b8e..8e11d72 100644
--- a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
+++ b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
@@ -5,18 +5,14 @@
 from hypothesis import strategies as st, assume
 import numpy as np
 
-from src.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
-from src.envs import BasicClusterEnv
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs import BasicClusterEnv, DifferentInPendingJobsRewardCaculator
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.metric_based.observation import (
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
+from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
-from src.envs.cluster_simulator.base.extractors.reward import (
-    DifferentInPendingJobsRewardCaculator,
-)
-from src.wrappers.cluster_simulator.dilation_wrapper import DilatorWrapper
 from tests.strategies.cluster_strategies import MetricClusterStrategies
 from tests.strategies.dilation_strategies.proto import DilationStrategies, Dilator
 
diff --git a/tests/strategies/dilation_strategies/proto.py b/tests/strategies/dilation_strategies/proto.py
index e1f4713..29d243a 100644
--- a/tests/strategies/dilation_strategies/proto.py
+++ b/tests/strategies/dilation_strategies/proto.py
@@ -3,7 +3,7 @@
 
 from hypothesis.strategies import SearchStrategy
 
-from src.envs.cluster_simulator.base.internal.dilation import AbstractDilation
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import AbstractDilation
 
 Dilator = TypeVar("Dilator", bound=AbstractDilation)
 
diff --git a/tests/strategies/env_strategies/basic_env_st.py b/tests/strategies/env_strategies/basic_env_st.py
index 7e94c4a..501421d 100644
--- a/tests/strategies/env_strategies/basic_env_st.py
+++ b/tests/strategies/env_strategies/basic_env_st.py
@@ -1,22 +1,22 @@
 from hypothesis import strategies as st, assume
 from typing import TYPE_CHECKING, TypedDict
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.single_slot import SingleSlotCluster
-from src.envs import BasicClusterEnv
-from src.envs.cluster_simulator.base.extractors.information import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.single_slot import SingleSlotCluster
+from src.server_simulator.envs import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.envs.cluster_simulator.deep_rm.observation import (
+from src.server_simulator.envs.cluster_simulator.deep_rm.creator import (
     DeepRMObservationCreator,
 )
-from src.envs.cluster_simulator.metric_based.observation import (
+from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
-from src.envs.cluster_simulator.single_slot.observation import (
+from src.server_simulator.envs.cluster_simulator.single_slot.observation import (
     SingleSlotObservationCreator,
 )
-from src.envs.cluster_simulator.base.extractors.reward import (
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
     DifferentInPendingJobsRewardCaculator,
 )
 from tests.strategies.cluster_strategies import (
diff --git a/tests/test_e2e/test_e2e_deeprm.py b/tests/test_e2e/test_e2e_deeprm.py
index 948ccb3..237defa 100644
--- a/tests/test_e2e/test_e2e_deeprm.py
+++ b/tests/test_e2e/test_e2e_deeprm.py
@@ -1,6 +1,7 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-from src import envs  # noqa: F401
+import src.server_simulator.envs # noqa: F401
+
 
 env = gym.make("ClusterScheduling-deeprm-v1").unwrapped
 check_env(env)
diff --git a/tests/test_e2e/test_e2e_matric_based.py b/tests/test_e2e/test_e2e_matric_based.py
index 349f452..f41942b 100644
--- a/tests/test_e2e/test_e2e_matric_based.py
+++ b/tests/test_e2e/test_e2e_matric_based.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-from src import envs  # noqa: F401
+import src.server_simulator.envs # noqa: F401
 
 env = gym.make("ClusterScheduling-metric-online-v1").unwrapped
 check_env(env)
diff --git a/tests/test_e2e/test_e2e_single_slot.py b/tests/test_e2e/test_e2e_single_slot.py
index 42cc7b2..3645500 100644
--- a/tests/test_e2e/test_e2e_single_slot.py
+++ b/tests/test_e2e/test_e2e_single_slot.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-from src import envs  # noqa: F401
+import src.server_simulator
 
 env = gym.make("ClusterScheduling-single-slot-v1").unwrapped
 check_env(env)
diff --git a/tests/test_envs/test_basic_env.py b/tests/test_envs/test_basic_env.py
index 58e94a3..6d8b2bb 100644
--- a/tests/test_envs/test_basic_env.py
+++ b/tests/test_envs/test_basic_env.py
@@ -3,15 +3,15 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.basic import BasicClusterEnv
+from src.server_simulator.envs import BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.actions import EnvironmentAction
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 from hypothesis import given, settings
 
-from src.envs.cluster_simulator.base.extractors.information import ClusterInformation
-from src.envs.cluster_simulator.base.extractors.observation import ClusterObservation
-from src.scheduler.random_scheduler import RandomScheduler
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
+from src.server_simulator.envs.cluster_simulator.base.renderer import ClusterObservation
+from src.server_simulator.scheduler import RandomScheduler
 from tests.strategies.env_strategies.basic_env_st import BasicGymEnvironmentStrategies
-from src.envs.cluster_simulator.basic import EnvironmentAction
 
 
 @given(env=BasicGymEnvironmentStrategies.creation())
diff --git a/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py b/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
index 4be6692..04e1c1d 100644
--- a/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
+++ b/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
@@ -1,9 +1,8 @@
-from src.envs.cluster_simulator.deep_rm.internal.jobs import DeepRMJobsConvertor
-from src.envs.cluster_simulator.deep_rm.internal.machines import DeepRMMachinesConvertor
-from src.scheduler.random_scheduler import RandomScheduler
+from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMJobsConvertor, DeepRMMachinesConvertor, \
+    DeepRMCluster, DeepRMCreators
+from src.server_simulator.scheduler import RandomScheduler
 from hypothesis import given, strategies as st, assume, settings, HealthCheck
-from src.envs.cluster_simulator.deep_rm import DeepRMCreators, DeepRMCluster
-from src.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 import numpy as np
 import pytest
 
diff --git a/tests/test_envs/test_cluster_simulator/test_metric_based/test_metric_cluster.py b/tests/test_envs/test_cluster_simulator/test_metric_based/test_metric_cluster.py
index 4e56c05..fdf45b3 100644
--- a/tests/test_envs/test_cluster_simulator/test_metric_based/test_metric_cluster.py
+++ b/tests/test_envs/test_cluster_simulator/test_metric_based/test_metric_cluster.py
@@ -1,9 +1,9 @@
 import pytest
 
-from src.envs.cluster_simulator.base.internal.job import Status
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 import numpy as np
 
-from src.envs.cluster_simulator.metric_based import (
+from src.server_simulator.envs.cluster_simulator.metric_based import (
     MetricClusterCreator,
     MetricCluster,
     MetricJobsConvertor,
@@ -11,7 +11,7 @@
 )
 from hypothesis import given, strategies as st, assume, settings, HealthCheck
 
-from src.scheduler.random_scheduler import RandomScheduler
+from src.server_simulator.scheduler import RandomScheduler
 from tests.strategies.cluster_strategies import MetricClusterStrategies
 from tests.test_envs.test_cluster_simulator.test_single_slot.test_single_slot_cluster import (
     seed_strategy,
diff --git a/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py b/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
index fc921bc..5a82feb 100644
--- a/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
+++ b/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
@@ -1,6 +1,6 @@
-from src.envs.cluster_simulator.base.internal.dilation import DilationState
-from src.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
-from src.envs.cluster_simulator.utils import array_operations
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import DilationState
+from src.server_simulator.envs.cluster_simulator import MetricBasedDilator
+from src.server_simulator.envs.cluster_simulator.utils import array_operations
 import numpy as np
 from hypothesis import given, strategies as st, assume, settings, HealthCheck
 
diff --git a/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py b/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
index aebf3db..5b3982f 100644
--- a/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
+++ b/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
@@ -1,9 +1,9 @@
 import numpy as np
 import pytest
 
-from src.envs.cluster_simulator.base.internal.job import Status as JobStatus
-from src.envs.cluster_simulator.single_slot.internal.jobs import SingleSlotJobsConvertor
-from src.envs.cluster_simulator.single_slot.internal.machines import (
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status as JobStatus
+from src.server_simulator.envs.cluster_simulator.single_slot.internal.jobs import SingleSlotJobsConvertor
+from src.server_simulator.envs.cluster_simulator.single_slot.internal.machines import (
     SingleSlotMachinesConvertor,
 )
 from tests.test_envs.test_cluster_simulator.test_single_slot.utils import (
diff --git a/tests/test_envs/test_cluster_simulator/test_single_slot/utils.py b/tests/test_envs/test_cluster_simulator/test_single_slot/utils.py
index a8b4560..4ef7a16 100644
--- a/tests/test_envs/test_cluster_simulator/test_single_slot/utils.py
+++ b/tests/test_envs/test_cluster_simulator/test_single_slot/utils.py
@@ -1,4 +1,4 @@
-from src.envs.cluster_simulator.single_slot import (
+from src.server_simulator.envs.cluster_simulator.single_slot import (
     SingleSlotCluster,
     SingleSlotClusterCreators,
 )
diff --git a/tests/test_envs/test_cluster_simulator/test_utils/test_array_operations.py b/tests/test_envs/test_cluster_simulator/test_utils/test_array_operations.py
index ee55dc3..df840c1 100644
--- a/tests/test_envs/test_cluster_simulator/test_utils/test_array_operations.py
+++ b/tests/test_envs/test_cluster_simulator/test_utils/test_array_operations.py
@@ -6,7 +6,7 @@
 from hypothesis import given, assume, HealthCheck, settings
 import numpy.typing as npt
 import typing as tp
-from src.envs.cluster_simulator.utils import array_operations
+from src.server_simulator.envs.cluster_simulator.utils import array_operations
 
 kernel_strategy = st.tuples(
     st.integers(min_value=2, max_value=5), st.integers(min_value=2, max_value=5)
diff --git a/tests/test_wrappers/test_dilation_wrapper.py b/tests/test_wrappers/test_dilation_wrapper.py
index bd5c181..8f396ec 100644
--- a/tests/test_wrappers/test_dilation_wrapper.py
+++ b/tests/test_wrappers/test_dilation_wrapper.py
@@ -3,23 +3,21 @@
 
 import numpy as np
 
-from src.envs.cluster_simulator.base.internal.job import Status
-from src.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
-from src.wrappers.cluster_simulator.dilation_wrapper import (
-    DilatorWrapper,
-    DilationEnvironmentAction,
-)
+from src.server_simulator.envs.cluster_simulator.actions import DilationEnvironmentAction
+from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 from typing import Tuple, Type
 from hypothesis import given, settings, HealthCheck, assume
-from src.envs.cluster_simulator.base.internal.dilation import (
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
     AbstractDilation,
     DilationState,
 )
-from src.scheduler.random_scheduler import RandomScheduler
+from src.server_simulator.envs.cluster_simulator.metric_based import MetricMachines
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
+from src.server_simulator.scheduler import RandomScheduler
+from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import DilatorWrapper
 from tests.strategies.dilation_strategies.metric_cluster_dilator_st import (
     MetricClusterDilationStrategies,
 )
-from src.envs.cluster_simulator.metric_based import MetricMachines
 import numpy.typing as npt
 
 DILATOR_CLASS_OPTIONS: Tuple[Type[AbstractDilation], ...] = (MetricBasedDilator,)

From c7247e2886f93a0a36dcf1ae91d35ab91d4ed021 Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 15:40:02 +0200
Subject: [PATCH 2/6] add import

---
 .../dilation_strategies/metric_cluster_dilator_st.py           | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
index 8e11d72..c74d3fc 100644
--- a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
+++ b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
@@ -13,6 +13,7 @@
 from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
+from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import DilatorWrapper
 from tests.strategies.cluster_strategies import MetricClusterStrategies
 from tests.strategies.dilation_strategies.proto import DilationStrategies, Dilator
 
@@ -66,6 +67,6 @@ def creation(draw) -> SearchStrategy[Dilator]:
         assume(kernel[0] > 1 and kernel[1] > 1)
         params = draw(MetricClusterDilationStrategies.initialization_parameters())
         params["kernel"] = kernel
-        return DilatorWrapper(  # type: ignore
+        return DilatorWrapper(   # type: ignore
             base_env, dilator_cls=MetricBasedDilator, **params
         )

From 23904d98b2041c37af943fc90ce2a0fe54b6af94 Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 15:41:12 +0200
Subject: [PATCH 3/6] add import

---
 tests/test_e2e/test_e2e_single_slot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_e2e/test_e2e_single_slot.py b/tests/test_e2e/test_e2e_single_slot.py
index 3645500..5f5198d 100644
--- a/tests/test_e2e/test_e2e_single_slot.py
+++ b/tests/test_e2e/test_e2e_single_slot.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-import src.server_simulator
+import src.server_simulator # noqa: F401
 
 env = gym.make("ClusterScheduling-single-slot-v1").unwrapped
 check_env(env)

From ee3051430c308e4a601e2a658c9a69b2504b676f Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 15:43:41 +0200
Subject: [PATCH 4/6] add import

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index d301f30..7b0c9bf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ log_cli_date_format = "%Y-%m-%d %H:%M:%S"
 line-length = 88
 
 [tool.hatch.build.targets.wheel]
-packages = ["src"]
+packages = ["src/server_simulator"]
 
 [build-system]
 requires = ["hatchling"]

From f38bbb8d06df6729dccb58b8181b6ce20a273c83 Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 15:45:41 +0200
Subject: [PATCH 5/6] add import

---
 src/server_simulator/envs/__init__.py             |  4 +++-
 .../envs/cluster_simulator/actions.py             |  5 ++++-
 .../base/extractors/information.py                |  4 +++-
 .../base/extractors/observation.py                |  8 ++++++--
 .../cluster_simulator/base/extractors/reward.py   |  4 +++-
 .../cluster_simulator/base/internal/cluster.py    | 14 +++++++++++---
 .../envs/cluster_simulator/basic.py               |  9 +++++++--
 .../envs/cluster_simulator/deep_rm/creator.py     | 13 ++++++++++---
 .../cluster_simulator/metric_based/creator.py     |  9 +++++++--
 .../cluster_simulator/metric_based/observation.py |  4 +++-
 .../cluster_simulator/metric_based/renderer.py    |  4 +++-
 .../envs/cluster_simulator/single_slot/creator.py |  4 +++-
 src/server_simulator/scheduler/base_scheduler.py  | 11 +++++++++--
 .../first_come_first_served_scheduler.py          |  4 +++-
 .../scheduler/random_scheduler.py                 |  4 +++-
 .../scheduler/round_robin_scheduler.py            |  4 +++-
 .../scheduler/shortest_job_first_scheduler.py     |  4 +++-
 .../cluster_simulator/dilation_wrapper.py         | 13 ++++++++++---
 .../wrappers/cluster_simulator/render_wrapper.py  |  4 +++-
 tests/strategies/cluster_strategies/deeprm_st.py  |  5 ++++-
 tests/strategies/cluster_strategies/metric_st.py  |  5 ++++-
 tests/strategies/cluster_strategies/proto.py      |  4 +++-
 .../metric_cluster_dilator_st.py                  | 15 +++++++++++----
 tests/strategies/dilation_strategies/proto.py     |  4 +++-
 tests/test_e2e/test_e2e_deeprm.py                 |  2 +-
 tests/test_e2e/test_e2e_matric_based.py           |  2 +-
 tests/test_e2e/test_e2e_single_slot.py            |  2 +-
 tests/test_envs/test_basic_env.py                 |  4 +++-
 .../test_deep_rm/test_deeprm_cluster.py           |  8 ++++++--
 .../test_metric_based/tests_metric_dilation.py    |  4 +++-
 .../test_single_slot/test_single_slot_cluster.py  |  8 ++++++--
 tests/test_wrappers/test_dilation_wrapper.py      | 12 +++++++++---
 32 files changed, 152 insertions(+), 49 deletions(-)

diff --git a/src/server_simulator/envs/__init__.py b/src/server_simulator/envs/__init__.py
index 537a7f7..bc9caba 100644
--- a/src/server_simulator/envs/__init__.py
+++ b/src/server_simulator/envs/__init__.py
@@ -1,7 +1,9 @@
 from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
     DifferentInPendingJobsRewardCaculator,
 )
-from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv as BasicClusterEnv
+from src.server_simulator.envs.cluster_simulator.basic import (
+    BasicClusterEnv as BasicClusterEnv,
+)
 from gymnasium import register
 
 from src.server_simulator.envs.cluster_simulator.deep_rm.creator import (
diff --git a/src/server_simulator/envs/cluster_simulator/actions.py b/src/server_simulator/envs/cluster_simulator/actions.py
index f67eb99..6b9d6f2 100644
--- a/src/server_simulator/envs/cluster_simulator/actions.py
+++ b/src/server_simulator/envs/cluster_simulator/actions.py
@@ -1,7 +1,10 @@
 from typing import NamedTuple, Tuple, TypeVar
 import gymnasium as gym
 
-from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterAction, ClusterABC
+from src.server_simulator.envs.cluster_simulator.base.internal.cluster import (
+    ClusterAction,
+    ClusterABC,
+)
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
 
diff --git a/src/server_simulator/envs/cluster_simulator/base/extractors/information.py b/src/server_simulator/envs/cluster_simulator/base/extractors/information.py
index 95f9ccc..ee213fa 100644
--- a/src/server_simulator/envs/cluster_simulator/base/extractors/information.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/information.py
@@ -2,7 +2,9 @@
 from typing import TypeVar, TypedDict, Generic
 import numpy.typing as npt
 
-from src.server_simulator.envs.cluster_simulator.base.extractors.observation import ClusterObservation
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
+    ClusterObservation,
+)
 
 
 class ClusterBaseInformation(TypedDict):
diff --git a/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py b/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
index e49171b..0176cb1 100644
--- a/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/observation.py
@@ -4,8 +4,12 @@
 import numpy.typing as npt
 
 from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
-from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollectionConvertor
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachinesCollectionConvertor
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
+    JobCollectionConvertor,
+)
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachinesCollectionConvertor,
+)
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
 MachinesRepresentation = TypeVar("MachinesRepresentation")
diff --git a/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py b/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
index 596f177..925160e 100644
--- a/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
+++ b/src/server_simulator/envs/cluster_simulator/base/extractors/reward.py
@@ -2,7 +2,9 @@
 import abc
 
 from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
-from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
+    ClusterInformation,
+)
 
 
 class RewardCaculator(Generic[ClusterInformation]):
diff --git a/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py b/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
index 060358c..677791b 100644
--- a/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
+++ b/src/server_simulator/envs/cluster_simulator/base/internal/cluster.py
@@ -3,9 +3,17 @@
 
 from rust_enum import enum, Case
 
-from src.server_simulator.envs.cluster_simulator.base.internal.job import Job, JobCollection
-from src.server_simulator.envs.cluster_simulator.base.internal.job import Status as JobStatus
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import Machine, MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
+    Job,
+    JobCollection,
+)
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
+    Status as JobStatus,
+)
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    Machine,
+    MachineCollection,
+)
 import logging
 
 T = tp.TypeVar("T")
diff --git a/src/server_simulator/envs/cluster_simulator/basic.py b/src/server_simulator/envs/cluster_simulator/basic.py
index 23516e1..7987ef2 100644
--- a/src/server_simulator/envs/cluster_simulator/basic.py
+++ b/src/server_simulator/envs/cluster_simulator/basic.py
@@ -2,8 +2,13 @@
 import typing as tp
 import numpy as np
 
-from src.server_simulator.envs.cluster_simulator.actions import EnvironmentAction, ActionConvertor
-from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.actions import (
+    EnvironmentAction,
+    ActionConvertor,
+)
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
+    RewardCaculator,
+)
 from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     ClusterInformation,
     BaceClusterInformationExtractor,
diff --git a/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py b/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
index 6b673dd..7c7914e 100644
--- a/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/deep_rm/creator.py
@@ -3,12 +3,19 @@
 from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
+    RewardCaculator,
+)
 from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
-from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMCluster, DeepRMCreators
-from src.server_simulator.envs.cluster_simulator.deep_rm.observation import DeepRMObservationCreator
+from src.server_simulator.envs.cluster_simulator.deep_rm import (
+    DeepRMCluster,
+    DeepRMCreators,
+)
+from src.server_simulator.envs.cluster_simulator.deep_rm.observation import (
+    DeepRMObservationCreator,
+)
 
 __all__ = ["DeepRMCreatorParameters", "DeepRMEnvCreator"]
 
diff --git a/src/server_simulator/envs/cluster_simulator/metric_based/creator.py b/src/server_simulator/envs/cluster_simulator/metric_based/creator.py
index 69a310f..042d041 100644
--- a/src/server_simulator/envs/cluster_simulator/metric_based/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/creator.py
@@ -3,12 +3,17 @@
 from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
+    RewardCaculator,
+)
 from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
 
-from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
+from src.server_simulator.envs.cluster_simulator.metric_based import (
+    MetricCluster,
+    MetricClusterCreator,
+)
 from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
diff --git a/src/server_simulator/envs/cluster_simulator/metric_based/observation.py b/src/server_simulator/envs/cluster_simulator/metric_based/observation.py
index 75c39d6..ff4a0df 100644
--- a/src/server_simulator/envs/cluster_simulator/metric_based/observation.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/observation.py
@@ -10,7 +10,9 @@
 import numpy as np
 
 from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster
-from src.server_simulator.envs.cluster_simulator.metric_based.internal.jobs import MetricJobsConvertor
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.jobs import (
+    MetricJobsConvertor,
+)
 from src.server_simulator.envs.cluster_simulator.metric_based.internal.machines import (
     MetricMachinesConvertor,
 )
diff --git a/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py b/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
index 07f084a..707374b 100644
--- a/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
+++ b/src/server_simulator/envs/cluster_simulator/metric_based/renderer.py
@@ -2,7 +2,9 @@
 
 from pygame import Color
 
-from src.server_simulator.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
+from src.server_simulator.envs.cluster_simulator.base.renderer import (
+    AbstractClusterGameRenderer,
+)
 from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservation,
 )
diff --git a/src/server_simulator/envs/cluster_simulator/single_slot/creator.py b/src/server_simulator/envs/cluster_simulator/single_slot/creator.py
index acdc963..4c010b5 100644
--- a/src/server_simulator/envs/cluster_simulator/single_slot/creator.py
+++ b/src/server_simulator/envs/cluster_simulator/single_slot/creator.py
@@ -3,7 +3,9 @@
 from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.server_simulator.envs.cluster_simulator.base.extractors.reward import RewardCaculator
+from src.server_simulator.envs.cluster_simulator.base.extractors.reward import (
+    RewardCaculator,
+)
 from src.server_simulator.envs.cluster_simulator.basic import BasicClusterEnv
 from typing import TypedDict, Optional
 from typing_extensions import Unpack
diff --git a/src/server_simulator/scheduler/base_scheduler.py b/src/server_simulator/scheduler/base_scheduler.py
index f191502..71fa4a8 100644
--- a/src/server_simulator/scheduler/base_scheduler.py
+++ b/src/server_simulator/scheduler/base_scheduler.py
@@ -2,8 +2,15 @@
 import typing as tp
 import abc
 
-from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection, Job, Status
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection, Machine
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
+    JobCollection,
+    Job,
+    Status,
+)
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachineCollection,
+    Machine,
+)
 
 T = tp.TypeVar("T")
 MachineT = tp.TypeVar("MachineT", bound=Machine)
diff --git a/src/server_simulator/scheduler/first_come_first_served_scheduler.py b/src/server_simulator/scheduler/first_come_first_served_scheduler.py
index 1847bd0..ed0e0f4 100644
--- a/src/server_simulator/scheduler/first_come_first_served_scheduler.py
+++ b/src/server_simulator/scheduler/first_come_first_served_scheduler.py
@@ -1,7 +1,9 @@
 import typing as tp
 
 from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachineCollection,
+)
 from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
diff --git a/src/server_simulator/scheduler/random_scheduler.py b/src/server_simulator/scheduler/random_scheduler.py
index f48cac0..86e1542 100644
--- a/src/server_simulator/scheduler/random_scheduler.py
+++ b/src/server_simulator/scheduler/random_scheduler.py
@@ -3,7 +3,9 @@
 import numpy as np
 
 from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachineCollection,
+)
 from src.server_simulator.scheduler.base_scheduler import ABCScheduler, T
 
 
diff --git a/src/server_simulator/scheduler/round_robin_scheduler.py b/src/server_simulator/scheduler/round_robin_scheduler.py
index 330f0e8..e98291e 100644
--- a/src/server_simulator/scheduler/round_robin_scheduler.py
+++ b/src/server_simulator/scheduler/round_robin_scheduler.py
@@ -1,7 +1,9 @@
 import typing as tp
 
 from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachineCollection,
+)
 from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
diff --git a/src/server_simulator/scheduler/shortest_job_first_scheduler.py b/src/server_simulator/scheduler/shortest_job_first_scheduler.py
index 580e1bb..2751d96 100644
--- a/src/server_simulator/scheduler/shortest_job_first_scheduler.py
+++ b/src/server_simulator/scheduler/shortest_job_first_scheduler.py
@@ -1,7 +1,9 @@
 import typing as tp
 
 from src.server_simulator.envs.cluster_simulator.base.internal.job import JobCollection
-from src.server_simulator.envs.cluster_simulator.base.internal.machine import MachineCollection
+from src.server_simulator.envs.cluster_simulator.base.internal.machine import (
+    MachineCollection,
+)
 from src.server_simulator.scheduler.base_scheduler import ABCScheduler
 
 T = tp.TypeVar("T")
diff --git a/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py b/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
index d0a34d6..4ebadac 100644
--- a/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
+++ b/src/server_simulator/wrappers/cluster_simulator/dilation_wrapper.py
@@ -7,15 +7,22 @@
 import numpy as np
 
 from src.server_simulator.envs import BasicClusterEnv
-from src.server_simulator.envs.cluster_simulator.actions import EnvironmentAction, DilationEnvironmentAction
-from src.server_simulator.envs.cluster_simulator.base.extractors.observation import BaseClusterObservation
+from src.server_simulator.envs.cluster_simulator.actions import (
+    EnvironmentAction,
+    DilationEnvironmentAction,
+)
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
+    BaseClusterObservation,
+)
 from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
     AbstractDilation,
     DilationAction,
     DilationState,
     AbstractDilationParams,
 )
-from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
+    ClusterInformation,
+)
 
 EnvironmentObservation = TypeVar("EnvironmentObservation", bound=BaseClusterObservation)
 WrapperObservation = TypeVar("WrapperObservation", bound=BaseClusterObservation)
diff --git a/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py b/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
index 298a242..4f24ce5 100644
--- a/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
+++ b/src/server_simulator/wrappers/cluster_simulator/render_wrapper.py
@@ -11,7 +11,9 @@
 from src.server_simulator.envs.cluster_simulator import (
     BaseClusterObservation,
 )
-from src.server_simulator.envs.cluster_simulator.base.renderer import AbstractClusterGameRenderer
+from src.server_simulator.envs.cluster_simulator.base.renderer import (
+    AbstractClusterGameRenderer,
+)
 
 EnvironmentObservation = TypeVar("EnvironmentObservation", bound=BaseClusterObservation)
 WrapperObservation = TypeVar("WrapperObservation", bound=BaseClusterObservation)
diff --git a/tests/strategies/cluster_strategies/deeprm_st.py b/tests/strategies/cluster_strategies/deeprm_st.py
index 38b65d6..06652df 100644
--- a/tests/strategies/cluster_strategies/deeprm_st.py
+++ b/tests/strategies/cluster_strategies/deeprm_st.py
@@ -1,7 +1,10 @@
 from hypothesis import strategies as st
 from hypothesis.strategies import SearchStrategy
 
-from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMCluster, DeepRMCreators
+from src.server_simulator.envs.cluster_simulator.deep_rm import (
+    DeepRMCluster,
+    DeepRMCreators,
+)
 from tests.strategies.cluster_strategies.proto import ClusterStrategies
 
 
diff --git a/tests/strategies/cluster_strategies/metric_st.py b/tests/strategies/cluster_strategies/metric_st.py
index cb82f0a..9b03ebf 100644
--- a/tests/strategies/cluster_strategies/metric_st.py
+++ b/tests/strategies/cluster_strategies/metric_st.py
@@ -1,7 +1,10 @@
 from hypothesis import strategies as st
 from hypothesis.strategies import SearchStrategy
 
-from src.server_simulator.envs.cluster_simulator.metric_based import MetricCluster, MetricClusterCreator
+from src.server_simulator.envs.cluster_simulator.metric_based import (
+    MetricCluster,
+    MetricClusterCreator,
+)
 from tests.strategies.cluster_strategies.proto import ClusterStrategies
 
 
diff --git a/tests/strategies/cluster_strategies/proto.py b/tests/strategies/cluster_strategies/proto.py
index e5395f9..48c0580 100644
--- a/tests/strategies/cluster_strategies/proto.py
+++ b/tests/strategies/cluster_strategies/proto.py
@@ -3,7 +3,9 @@
 
 from hypothesis.strategies import SearchStrategy
 
-from src.server_simulator.envs.cluster_simulator.base.extractors.observation import BaseObservationCreatorProtocol
+from src.server_simulator.envs.cluster_simulator.base.extractors.observation import (
+    BaseObservationCreatorProtocol,
+)
 from src.server_simulator.envs.cluster_simulator.base.internal.cluster import ClusterABC
 
 Cluster = TypeVar("Cluster", bound=ClusterABC)
diff --git a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
index c74d3fc..2642dd6 100644
--- a/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
+++ b/tests/strategies/dilation_strategies/metric_cluster_dilator_st.py
@@ -5,15 +5,22 @@
 from hypothesis import strategies as st, assume
 import numpy as np
 
-from src.server_simulator.envs import BasicClusterEnv, DifferentInPendingJobsRewardCaculator
+from src.server_simulator.envs import (
+    BasicClusterEnv,
+    DifferentInPendingJobsRewardCaculator,
+)
 from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
     BaceClusterInformationExtractor,
 )
-from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import (
+    MetricBasedDilator,
+)
 from src.server_simulator.envs.cluster_simulator.metric_based.observation import (
     MetricClusterObservationCreator,
 )
-from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import DilatorWrapper
+from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import (
+    DilatorWrapper,
+)
 from tests.strategies.cluster_strategies import MetricClusterStrategies
 from tests.strategies.dilation_strategies.proto import DilationStrategies, Dilator
 
@@ -67,6 +74,6 @@ def creation(draw) -> SearchStrategy[Dilator]:
         assume(kernel[0] > 1 and kernel[1] > 1)
         params = draw(MetricClusterDilationStrategies.initialization_parameters())
         params["kernel"] = kernel
-        return DilatorWrapper(   # type: ignore
+        return DilatorWrapper(  # type: ignore
             base_env, dilator_cls=MetricBasedDilator, **params
         )
diff --git a/tests/strategies/dilation_strategies/proto.py b/tests/strategies/dilation_strategies/proto.py
index 29d243a..4018418 100644
--- a/tests/strategies/dilation_strategies/proto.py
+++ b/tests/strategies/dilation_strategies/proto.py
@@ -3,7 +3,9 @@
 
 from hypothesis.strategies import SearchStrategy
 
-from src.server_simulator.envs.cluster_simulator.base.internal.dilation import AbstractDilation
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
+    AbstractDilation,
+)
 
 Dilator = TypeVar("Dilator", bound=AbstractDilation)
 
diff --git a/tests/test_e2e/test_e2e_deeprm.py b/tests/test_e2e/test_e2e_deeprm.py
index 237defa..d7b880d 100644
--- a/tests/test_e2e/test_e2e_deeprm.py
+++ b/tests/test_e2e/test_e2e_deeprm.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-import src.server_simulator.envs # noqa: F401
+import src.server_simulator.envs  # noqa: F401
 
 
 env = gym.make("ClusterScheduling-deeprm-v1").unwrapped
diff --git a/tests/test_e2e/test_e2e_matric_based.py b/tests/test_e2e/test_e2e_matric_based.py
index f41942b..11909e8 100644
--- a/tests/test_e2e/test_e2e_matric_based.py
+++ b/tests/test_e2e/test_e2e_matric_based.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-import src.server_simulator.envs # noqa: F401
+import src.server_simulator.envs  # noqa: F401
 
 env = gym.make("ClusterScheduling-metric-online-v1").unwrapped
 check_env(env)
diff --git a/tests/test_e2e/test_e2e_single_slot.py b/tests/test_e2e/test_e2e_single_slot.py
index 5f5198d..57c9532 100644
--- a/tests/test_e2e/test_e2e_single_slot.py
+++ b/tests/test_e2e/test_e2e_single_slot.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 from gymnasium.utils.env_checker import check_env
-import src.server_simulator # noqa: F401
+import src.server_simulator  # noqa: F401
 
 env = gym.make("ClusterScheduling-single-slot-v1").unwrapped
 check_env(env)
diff --git a/tests/test_envs/test_basic_env.py b/tests/test_envs/test_basic_env.py
index 6d8b2bb..87c9a99 100644
--- a/tests/test_envs/test_basic_env.py
+++ b/tests/test_envs/test_basic_env.py
@@ -8,7 +8,9 @@
 from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 from hypothesis import given, settings
 
-from src.server_simulator.envs.cluster_simulator.base.extractors.information import ClusterInformation
+from src.server_simulator.envs.cluster_simulator.base.extractors.information import (
+    ClusterInformation,
+)
 from src.server_simulator.envs.cluster_simulator.base.renderer import ClusterObservation
 from src.server_simulator.scheduler import RandomScheduler
 from tests.strategies.env_strategies.basic_env_st import BasicGymEnvironmentStrategies
diff --git a/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py b/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
index 04e1c1d..bfc79a7 100644
--- a/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
+++ b/tests/test_envs/test_cluster_simulator/test_deep_rm/test_deeprm_cluster.py
@@ -1,5 +1,9 @@
-from src.server_simulator.envs.cluster_simulator.deep_rm import DeepRMJobsConvertor, DeepRMMachinesConvertor, \
-    DeepRMCluster, DeepRMCreators
+from src.server_simulator.envs.cluster_simulator.deep_rm import (
+    DeepRMJobsConvertor,
+    DeepRMMachinesConvertor,
+    DeepRMCluster,
+    DeepRMCreators,
+)
 from src.server_simulator.scheduler import RandomScheduler
 from hypothesis import given, strategies as st, assume, settings, HealthCheck
 from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
diff --git a/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py b/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
index 5a82feb..0a53fbd 100644
--- a/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
+++ b/tests/test_envs/test_cluster_simulator/test_metric_based/tests_metric_dilation.py
@@ -1,4 +1,6 @@
-from src.server_simulator.envs.cluster_simulator.base.internal.dilation import DilationState
+from src.server_simulator.envs.cluster_simulator.base.internal.dilation import (
+    DilationState,
+)
 from src.server_simulator.envs.cluster_simulator import MetricBasedDilator
 from src.server_simulator.envs.cluster_simulator.utils import array_operations
 import numpy as np
diff --git a/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py b/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
index 5b3982f..f5ae101 100644
--- a/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
+++ b/tests/test_envs/test_cluster_simulator/test_single_slot/test_single_slot_cluster.py
@@ -1,8 +1,12 @@
 import numpy as np
 import pytest
 
-from src.server_simulator.envs.cluster_simulator.base.internal.job import Status as JobStatus
-from src.server_simulator.envs.cluster_simulator.single_slot.internal.jobs import SingleSlotJobsConvertor
+from src.server_simulator.envs.cluster_simulator.base.internal.job import (
+    Status as JobStatus,
+)
+from src.server_simulator.envs.cluster_simulator.single_slot.internal.jobs import (
+    SingleSlotJobsConvertor,
+)
 from src.server_simulator.envs.cluster_simulator.single_slot.internal.machines import (
     SingleSlotMachinesConvertor,
 )
diff --git a/tests/test_wrappers/test_dilation_wrapper.py b/tests/test_wrappers/test_dilation_wrapper.py
index 8f396ec..13ade54 100644
--- a/tests/test_wrappers/test_dilation_wrapper.py
+++ b/tests/test_wrappers/test_dilation_wrapper.py
@@ -3,7 +3,9 @@
 
 import numpy as np
 
-from src.server_simulator.envs.cluster_simulator.actions import DilationEnvironmentAction
+from src.server_simulator.envs.cluster_simulator.actions import (
+    DilationEnvironmentAction,
+)
 from src.server_simulator.envs.cluster_simulator.base.internal.job import Status
 from typing import Tuple, Type
 from hypothesis import given, settings, HealthCheck, assume
@@ -12,9 +14,13 @@
     DilationState,
 )
 from src.server_simulator.envs.cluster_simulator.metric_based import MetricMachines
-from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import MetricBasedDilator
+from src.server_simulator.envs.cluster_simulator.metric_based.internal.dilation import (
+    MetricBasedDilator,
+)
 from src.server_simulator.scheduler import RandomScheduler
-from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import DilatorWrapper
+from src.server_simulator.wrappers.cluster_simulator.dilation_wrapper import (
+    DilatorWrapper,
+)
 from tests.strategies.dilation_strategies.metric_cluster_dilator_st import (
     MetricClusterDilationStrategies,
 )

From 7774f26223ff5a2eb659eaf755e2811a14a8d6a2 Mon Sep 17 00:00:00 2001
From: dev0Guy <97923827+dev0Guy@users.noreply.github.com>
Date: Sat, 21 Mar 2026 16:03:03 +0200
Subject: [PATCH 6/6] add import

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7b0c9bf..a064895 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "server-simulator"
-version = "0.1.3"
+version = "0.1.4"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"