Version bump to v0.30.0.dev0 (#3770)

Author: rithwik-db

composer/_version.py

@@ -3,4 +3,4 @@
 
 """The Composer Version."""
 
-__version__ = '0.29.0.dev0'
+__version__ = '0.30.0.dev0'

composer/core/data_spec.py

@@ -14,7 +14,7 @@
 import torch.utils.data
 from torch.utils.data.distributed import DistributedSampler
 
-from composer.utils import VersionedDeprecationWarning, dist, ensure_tuple
+from composer.utils import dist, ensure_tuple
 
 if TYPE_CHECKING:
     from composer.core.types import Batch
@@ -155,10 +155,6 @@ class DataSpec:
         num_tokens (int, optional): The total number of tokens in an epoch. This field is used by the
             :class:`.Timestamp` (training progress tracker).
 
-        device_transforms ((Batch) -> Batch, optional): Deprecated argument. Please use ``batch_transforms`` for batch
-            level transformations on CPU and ``microbatch_transforms`` for microbatch level transformations on target
-            device.
-
         batch_transforms ((Batch) -> Batch, optional): Function called by the :class:`.Trainer` to modify the
             batch before it is moved onto the device. For example, this function can be used for CPU-based
             normalization. It can modify the batch in-place, and it should return the modified batch. If not specified,
@@ -194,7 +190,6 @@ def __init__(
         dataloader: Union[Iterable, torch.utils.data.DataLoader],
         num_samples: Optional[int] = None,
         num_tokens: Optional[int] = None,
-        device_transforms: Optional[Callable[[Batch], Batch]] = None,
         batch_transforms: Optional[Callable[[Batch], Batch]] = None,
         microbatch_transforms: Optional[Callable[[Batch], Batch]] = None,
         split_batch: Optional[Callable[[Batch, Union[int, float]], Sequence[Batch]]] = None,
@@ -203,22 +198,6 @@ def __init__(
     ) -> None:
         self.dataloader: Union[Iterable, torch.utils.data.DataLoader] = dataloader
         self.num_tokens = num_tokens
-        if device_transforms is not None:
-            if batch_transforms is not None:
-                raise ValueError(
-                    'Cannot specify both `device_transforms` and `batch_transforms`. Please use `batch_transforms` for '
-                    'batch level transformations on CPU and `microbatch_transforms` for microbatch level transformations '
-                    'on target device.',
-                )
-            warnings.warn(
-                VersionedDeprecationWarning(
-                    'The `device_transforms` argument is deprecated. Please use `batch_transforms` for batch level '
-                    'transformations on CPU and `microbatch_transforms` for microbatch level transformations on target '
-                    'device.',
-                    'v0.29.0',
-                ),
-            )
-            self.batch_transforms = device_transforms
         self.batch_transforms = self._default_transforms if batch_transforms is None else batch_transforms
         self.microbatch_transforms = self._default_transforms if microbatch_transforms is None else microbatch_transforms
         self.split_batch = default_split_batch if split_batch is None else split_batch

docker/README.md

@@ -15,8 +15,8 @@ all dependencies for both NLP and Vision models. They are built on top of the
 <!-- BEGIN_COMPOSER_BUILD_MATRIX -->
 | Composer Version   | CUDA Support   | Docker Tag                                                                                                                                                       |
 |--------------------|----------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| 0.28.0             | Yes            | `mosaicml/composer:latest`, `mosaicml/composer:0.28.0`                 |
-| 0.28.0             | No             | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.28.0_cpu` |
+| 0.29.0             | Yes            | `mosaicml/composer:latest`, `mosaicml/composer:0.29.0`                 |
+| 0.29.0             | No             | `mosaicml/composer:latest_cpu`, `mosaicml/composer:0.29.0_cpu` |
 <!-- END_COMPOSER_BUILD_MATRIX -->
 
 **Note**: For a lightweight installation, we recommended using a [MosaicML PyTorch Image](#pytorch-images) and manually

docker/build_matrix.yaml

@@ -142,36 +142,36 @@
   TORCHVISION_VERSION: 0.19.1
 - AWS_OFI_NCCL_VERSION: ''
   BASE_IMAGE: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
-  COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.28.0
+  COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.29.0
   CUDA_VERSION: 12.4.1
-  IMAGE_NAME: composer-0-28-0
+  IMAGE_NAME: composer-0-29-0
   MOFED_VERSION: latest-23.10
   NVIDIA_REQUIRE_CUDA_OVERRIDE: ''
   PYTHON_VERSION: '3.11'
   PYTORCH_NIGHTLY_URL: ''
   PYTORCH_NIGHTLY_VERSION: ''
   PYTORCH_VERSION: 2.6.0
   TAGS:
-  - mosaicml/composer:0.28.0
-  - ghcr.io/databricks-mosaic/composer:0.28.0
+  - mosaicml/composer:0.29.0
+  - ghcr.io/databricks-mosaic/composer:0.29.0
   - mosaicml/composer:latest
   - ghcr.io/databricks-mosaic/composer:latest
   TARGET: composer_stage
   TORCHVISION_VERSION: 0.21.0
 - AWS_OFI_NCCL_VERSION: ''
   BASE_IMAGE: ubuntu:22.04
-  COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.28.0
+  COMPOSER_INSTALL_COMMAND: mosaicml[all]==0.29.0
   CUDA_VERSION: ''
-  IMAGE_NAME: composer-0-28-0-cpu
+  IMAGE_NAME: composer-0-29-0-cpu
   MOFED_VERSION: latest-23.10
   NVIDIA_REQUIRE_CUDA_OVERRIDE: ''
   PYTHON_VERSION: '3.11'
   PYTORCH_NIGHTLY_URL: ''
   PYTORCH_NIGHTLY_VERSION: ''
   PYTORCH_VERSION: 2.6.0
   TAGS:
-  - mosaicml/composer:0.28.0_cpu
-  - ghcr.io/databricks-mosaic/composer:0.28.0_cpu
+  - mosaicml/composer:0.29.0_cpu
+  - ghcr.io/databricks-mosaic/composer:0.29.0_cpu
   - mosaicml/composer:latest_cpu
   - ghcr.io/databricks-mosaic/composer:latest_cpu
   TARGET: composer_stage

docker/generate_build_matrix.py

@@ -244,7 +244,7 @@ def _main():
     composer_entries = []
 
     # The `GIT_COMMIT` is a placeholder and Jenkins will substitute it with the actual git commit for the `composer_staging` images
-    composer_versions = ['0.28.0']  # Only build images for the latest composer version
+    composer_versions = ['0.29.0']  # Only build images for the latest composer version
     composer_python_versions = [PRODUCTION_PYTHON_VERSION]  # just build composer against the latest
 
     for product in itertools.product(composer_python_versions, composer_versions, cuda_options):