Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .idea/misc.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

3 changes: 2 additions & 1 deletion .idea/server-simulator.iml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -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"
Expand Down Expand Up @@ -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"]
Expand Down
12 changes: 7 additions & 5 deletions src/envs/__init__.py → src/server_simulator/envs/__init__.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,20 @@
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,
)
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,10 @@
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)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,9 @@
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):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,13 @@
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")
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
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]):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,17 @@

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")
Expand Down
Original file line number Diff line number Diff line change
@@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,22 @@
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)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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]):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,14 +1,21 @@
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"]

Expand Down
Empty file.
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down
Original file line number Diff line number Diff line change
@@ -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,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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]):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,15 +1,20 @@
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,
)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)
Expand Down
Loading
Loading