diff --git a/.github/mcp/mcp_pytest.py b/.github/mcp/mcp_pytest.py index 88922486c8..398f371ad0 100644 --- a/.github/mcp/mcp_pytest.py +++ b/.github/mcp/mcp_pytest.py @@ -28,7 +28,6 @@ 'integration_type': 'git_repo', 'git_repo': 'mosaicml/composer', 'ssh_clone': 'False', - 'pip_install': '--user -e .[all]', } if args.git_branch is not None and args.git_commit is None: git_integration['git_branch'] = args.git_branch @@ -49,6 +48,8 @@ command += f''' + pip install --user .[all] + export COMMON_ARGS="-v --durations=20 -m '{args.pytest_markers}'" make test PYTEST='{args.pytest_command}' EXTRA_ARGS="$COMMON_ARGS --codeblocks" diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index f17b013e00..eca4e6d3c9 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -16,7 +16,7 @@ repos: - repo: https://github.com/pycqa/isort hooks: - id: isort - rev: 5.10.1 + rev: 5.12.0 # - repo: https://github.com/pycqa/pylint # hooks: # - id: pylint diff --git a/setup.py b/setup.py index e0ea7e02d8..b79935ed66 100644 --- a/setup.py +++ b/setup.py @@ -180,7 +180,7 @@ def package_files(prefix: str, directory: str, extension: str): ] extra_deps['streaming'] = [ - 'mosaicml-streaming<0.3.*', + 'mosaicml-streaming<0.3', 'boto3>=1.21.45,<2', 'paramiko>=2.11.0,<3', ]