Renamed core to classes

Author: remi-or

src/transformers/generation/continuous_batching/__init__.py

@@ -14,7 +14,7 @@
 # limitations under the License.
 from .cache import PagedAttentionCache
 from .continuous_api import ContinuousBatchingManager, ContinuousMixin
-from .core import RequestState, RequestStatus
+from .classes import RequestState, RequestStatus
 
 
 __all__ = ["PagedAttentionCache", "RequestState", "RequestStatus", "ContinuousMixin", "ContinuousBatchingManager"]

src/transformers/generation/continuous_batching/cache.py

@@ -21,7 +21,7 @@
 from ...configuration_utils import PretrainedConfig
 from ...generation.configuration_utils import GenerationConfig
 from ...utils.metrics import attach_tracer, traced
-from .core import RequestState, get_device_and_memory_breakdown, logger
+from .classes import RequestState, get_device_and_memory_breakdown, logger
 
 
 @attach_tracer()

src/transformers/generation/continuous_batching/core.py renamed to src/transformers/generation/continuous_batching/classes.py

@@ -30,7 +30,7 @@
 from ...utils.logging import logging
 from ...utils.metrics import ContinuousBatchProcessorMetrics, attach_tracer, traced
 from .cache import PagedAttentionCache
-from .core import GenerationOutput, RequestState, RequestStatus, get_device_and_memory_breakdown, logger
+from .classes import GenerationOutput, RequestState, RequestStatus, get_device_and_memory_breakdown, logger
 from .scheduler import SCHEDULER_MAPPING, FIFOScheduler, Scheduler
 
 

src/transformers/generation/continuous_batching/continuous_api.py

@@ -17,7 +17,7 @@
 
 from ...utils.metrics import attach_tracer, traced
 from .cache import PagedAttentionCache
-from .core import RequestState, RequestStatus
+from .classes import RequestState, RequestStatus
 
 
 class Scheduler(ABC):