[Do not merge] Apply https://github.com/optuna/optuna/pull/5217 on https://github.com/HideakiImamura/optuna/pull/6

.github/workflows/checks-integration.yml

@@ -7,6 +7,7 @@ on:
   schedule:
     - cron: '0 23 * * SUN-THU'
   workflow_dispatch:
+  pull_request: {}
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref == 'refs/heads/master' && github.run_number || github.ref }}

.github/workflows/windows-tests.yml

@@ -73,7 +73,7 @@ jobs:
     - name: Tests
       if:  ${{ github.event_name != 'schedule' && github.event_name != 'workflow_dispatch' }}
       run: |
-        pytest -m "not integration and not slow"
+        pytest -m "not integration"
       env:
         MPLBACKEND: "QtAgg" # Use QtAgg as matplotlib backend.
 
@@ -118,6 +118,8 @@ jobs:
         pip install --progress-bar off .[optional]
         pip install --progress-bar off .[integration]
         pip install "distributed<2023.3.2"
+        # TODO(nabenabe0928): Remove the version constraint on Torch.
+        pip install "torch<2.2.0"
 
     - name: Output installed packages
       run: |

optuna/integration/pytorch_distributed.py

@@ -21,7 +21,7 @@
 with try_import() as _imports:
     import torch
     import torch.distributed as dist
-    from torch.distributed import ProcessGroup  # type: ignore[attr-defined]
+    from torch.distributed import ProcessGroup
 
 
 if TYPE_CHECKING:

tests/artifacts_tests/test_boto3.py

@@ -11,10 +11,10 @@
 
 
 try:
-    from moto import mock_aws
+    # TODO(nabenabe0928): Replace it with `from moto import mock_aws` after dropping Python3.7.
+    from moto import mock_aws  # type: ignore[attr-defined]
 except ImportError:
-    from moto import mock_s3 as mock_aws
-
+    from moto import mock_s3 as mock_aws  # type: ignore[attr-defined,no-redef]
 
 if TYPE_CHECKING:
     from collections.abc import Iterator