Fix LRSchedulerWrapper with shared memory address issue. (#135)

* fix: Fix LRSchedulerWrapper with shared memory address issue.

* feat: update workflows, test data, and remove torch.jit

* feat: lint the files.

* fix: Fix mypy error.

.github/workflows/ci.yml

@@ -59,7 +59,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9"]
+        python-version: ["3.10"]
 
     steps:
       - uses: actions/checkout@v2

tests/common/ckpt_test.py

@@ -1,4 +1,5 @@
 """Test checkpoint."""
+
 from vis4d.common.ckpt import get_torchvision_models, load_from_torchvision
 
 

tests/common/imports_test.py

@@ -1,4 +1,5 @@
 """Test imports."""
+
 import unittest
 
 from vis4d.common.imports import package_available

tests/common/logging_test.py

@@ -1,4 +1,5 @@
 """Test logging."""
+
 import logging
 import os
 import shutil

tests/common/named_tuple_test.py

@@ -1,4 +1,5 @@
 """Test named tuple."""
+
 import unittest
 from collections import namedtuple
 

tests/common/prettyprint_test.py

@@ -1,4 +1,5 @@
 """Test cases for pretty printing."""
+
 import unittest
 from typing import Any
 

tests/common/progress_test.py

@@ -1,4 +1,5 @@
 """Test cases for progress bar."""
+
 import unittest
 
 import torch

tests/common/time_test.py

@@ -1,4 +1,5 @@
 """Test cases for time."""
+
 import time
 import unittest
 from io import StringIO

tests/config/common/datasets/bdd100k/detect_test.py

@@ -1,4 +1,5 @@
 """Testcase for default data pipelines for detection."""
+
 import unittest
 
 from tests.util import get_test_data

tests/config/common/datasets/bdd100k/sem_seg_test.py

@@ -1,4 +1,5 @@
 """Testcase for default data pipelines for segmentation."""
+
 import unittest
 
 from tests.util import get_test_data

tests/config/common/datasets/coco/detection_test.py

@@ -1,4 +1,5 @@
 """Test cases for dataset and dataloader using COCO."""
+
 from __future__ import annotations
 
 import unittest

tests/config/common/datasets/nuscenes/nuscenes_mono_test.py

@@ -1,4 +1,5 @@
 """Test NuScenes monocular dataset config."""
+
 from __future__ import annotations
 
 import unittest

tests/config/common/datasets/nuscenes/nuscenes_test.py

@@ -1,4 +1,5 @@
 """Test NuScenes multi-sensor video dataset config."""
+
 from __future__ import annotations
 
 import unittest

tests/config/common/datasets/shift/test_shift_tasks.py

@@ -1,4 +1,5 @@
 """Test cases for dataset and dataloader using COCO."""
+
 from __future__ import annotations
 
 import unittest

tests/config/registry_test.py

@@ -1,4 +1,5 @@
 """Test config registry."""
+
 from __future__ import annotations
 
 import unittest

tests/config/show_connection_test.py

@@ -1,4 +1,5 @@
 """Show config connection tests."""
+
 import unittest
 
 from tests.zoo.util import content_equal, get_config_for_name
@@ -24,9 +25,7 @@ def test_show_frcnn(self) -> None:
         model = instantiate_classes(config.model)
 
         # Change the data root of evaluator callback to the test data
-        config.callbacks[
-            3
-        ].init_args.evaluator.init_args.data_root = (
+        config.callbacks[3].init_args.evaluator.init_args.data_root = (
             "tests/vis4d-test-data/coco_test"
         )
         config.callbacks[3].init_args.evaluator.init_args.split = "train"

tests/config/sweep/sweep_test.py

@@ -1,6 +1,5 @@
 """Vis4d Sweep Config Tests."""
 
-
 import unittest
 
 import numpy as np

tests/data/cbgs_test.py

@@ -1,4 +1,5 @@
 """Class-balanced Grouping and Sampling for 3D Object Detection test."""
+
 from tests.util import get_test_data
 from vis4d.data.cbgs import CBGSDataset
 from vis4d.data.datasets.nuscenes import NuScenes, nuscenes_class_map

tests/data/data_pipe_test.py

@@ -1,4 +1,5 @@
 """Test data pipelines."""
+
 import os
 
 from tests.util import get_test_data

tests/data/datasets/bdd100k_test.py

@@ -1,4 +1,5 @@
 """BDD100K dataset testing class."""
+
 import os
 import unittest
 

tests/data/datasets/coco_test.py

@@ -1,4 +1,5 @@
 """COCO dataset testing class."""
+
 import unittest
 
 import torch

tests/data/datasets/nuscenes_mono_test.py

@@ -1,4 +1,5 @@
 """NuScenes monocular dataset testing class."""
+
 import unittest
 
 from tests.util import get_test_data

tests/data/datasets/nuscenes_test.py

@@ -1,4 +1,5 @@
 """NuScenes dataset testing class."""
+
 import unittest
 
 from tests.util import get_test_data

tests/data/datasets/nuscenes_trajectory_test.py

@@ -1,4 +1,5 @@
 """NuScenes trajectory dataset tests."""
+
 import unittest
 
 from tests.util import get_test_data

tests/data/datasets/s3dis_test.py

@@ -1,4 +1,5 @@
 """S3DIS dataset testing class."""
+
 import unittest
 
 from tests.util import get_test_data

tests/data/datasets/scalabel_test.py

@@ -1,4 +1,5 @@
 """Tests for scalabel dataset."""
+
 import os
 
 import numpy as np

tests/data/datasets/shift_test.py

@@ -1,4 +1,5 @@
 """S3DIS dataset testing class."""
+
 import unittest
 
 import numpy as np

tests/data/datasets/torchvision_test.py

@@ -1,4 +1,5 @@
 """Test for Torchvision dataset wrapper."""
+
 import os
 import shutil
 import tempfile

tests/data/io/backend_test.py

@@ -1,4 +1,5 @@
 """Testcases for data backends."""
+
 import os
 import shutil
 import sys

tests/data/io/to_hdf5_test.py

@@ -1,4 +1,5 @@
 """Testcases for hdf5 dataset conversion."""
+
 import os
 import unittest
 

tests/data/iterable_test.py

@@ -1,4 +1,5 @@
 """Iterable datasets tests."""
+
 from __future__ import annotations
 
 import numpy as np

tests/data/loader_test.py

@@ -1,4 +1,5 @@
 """Test loader components."""
+
 import unittest
 
 import torch

tests/data/reference_test.py

@@ -1,4 +1,5 @@
 """Testcases for reference view sampling."""
+
 import unittest
 
 from vis4d.data.reference import SequentialViewSampler, UniformViewSampler

tests/data/resample_test.py

@@ -1,4 +1,5 @@
 """Resample index to recover the original dataset length test."""
+
 from tests.util import get_test_data
 from vis4d.data.datasets.nuscenes import NuScenes
 from vis4d.data.resample import ResampleDataset

tests/data/transforms/autoaugment_test.py

@@ -1,4 +1,5 @@
 """Test cases for auto-augment transform."""
+
 import numpy as np
 
 from vis4d.common.util import set_random_seed

tests/data/transforms/random_erasing_test.py

@@ -1,4 +1,5 @@
 """Test cases for normalize transform."""
+
 import numpy as np
 
 from vis4d.data.transforms.random_erasing import RandomErasing

tests/data/transforms/to_tensor_test.py

@@ -1,4 +1,5 @@
 """Test ToTensor transform."""
+
 import numpy as np
 import torch
 

tests/engine/callbacks/checkpoint_test.py

@@ -1,4 +1,5 @@
 """Test cases for checkpoint callback."""
+
 import shutil
 import tempfile
 import unittest

tests/engine/callbacks/ema_test.py

@@ -1,4 +1,5 @@
 """Test cases for EMA callback."""
+
 import unittest
 
 import torch

tests/engine/callbacks/evaluator_test.py

@@ -1,4 +1,5 @@
 """Test cases for evaluator callback."""
+
 import shutil
 import tempfile
 import unittest

tests/engine/callbacks/logging_test.py

@@ -1,4 +1,5 @@
 """Test cases for logging callback."""
+
 import unittest
 
 from tests.util import MOCKLOSS, MockModel

tests/engine/callbacks/visualizer_test.py

@@ -1,4 +1,5 @@
 """Test cases for visualizer callback."""
+
 import shutil
 import tempfile
 import unittest

tests/engine/connectors_test.py

@@ -1,4 +1,5 @@
 """Test data connector."""
+
 import torch
 
 from vis4d.engine.connectors import MultiSensorDataConnector, data_key

tests/engine/optim/optimizer_test.py

@@ -1,4 +1,5 @@
 """Test cases for optimizer."""
+
 from __future__ import annotations
 
 import unittest

tests/engine/optim/scheduler_test.py

@@ -1,4 +1,5 @@
 """Test cases for learning rate schedulers."""
+
 import copy
 import unittest
 

tests/engine/trainer_test.py

@@ -1,4 +1,5 @@
 """Engine trainer tests."""
+
 from __future__ import annotations
 
 import shutil

tests/engine/util_test.py

@@ -1,4 +1,5 @@
 """Test engine util."""
+
 from collections import namedtuple
 from dataclasses import dataclass
 

tests/eval/bdd100k/seg_test.py

@@ -1,4 +1,5 @@
 """Testcases for BDD100K segmentation evaluator."""
+
 from __future__ import annotations
 
 import os.path as osp

tests/eval/bdd100k/track_test.py

@@ -1,4 +1,5 @@
 """Testcases for BDD100K tracking evaluator."""
+
 from __future__ import annotations
 
 import os.path as osp

tests/eval/coco/detect_test.py

@@ -1,4 +1,5 @@
 """Testcases for COCO evaluator."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/common/binary_test.py

@@ -1,4 +1,5 @@
 """Occupancy test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/common/depth_test.py

@@ -1,4 +1,5 @@
 """Depth eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/common/flow_test.py

@@ -1,4 +1,5 @@
 """Depth eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/common/seg_test.py

@@ -1,4 +1,5 @@
 """Seg eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/kitti/depth_test.py

@@ -1,4 +1,5 @@
 """KITTI Depth Evaluation test."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/nuscenes/detect3d_test.py

@@ -1,4 +1,5 @@
 """Testcases for NuScenes 3D detection evaluator."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/nuscenes/track3d_test.py

@@ -1,4 +1,5 @@
 """Testcases for NuScenes 3D tracking evaluator."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/scalabel/detect_test.py

@@ -1,4 +1,5 @@
 """Testcases for BDD100K detection evaluator."""
+
 from __future__ import annotations
 
 import os.path as osp

tests/eval/scalabel/track_test.py

@@ -1,4 +1,5 @@
 """Testcases for BDD100K tracking evaluator."""
+
 from __future__ import annotations
 
 import os.path as osp

tests/eval/shift/depth_test.py

@@ -1,4 +1,5 @@
 """SHIFT eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/shift/detect_test.py

@@ -1,4 +1,5 @@
 """SHIFT eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/shift/flow_test.py

@@ -1,4 +1,5 @@
 """SHIFT eval test cases."""
+
 from __future__ import annotations
 
 import unittest

tests/eval/shift/multitask_writer_test.py

@@ -1,4 +1,5 @@
 """SHIFT eval test cases."""
+
 from __future__ import annotations
 
 import os

tests/eval/shift/seg_test.py

@@ -1,4 +1,5 @@
 """SHIFT eval test cases."""
+
 from __future__ import annotations
 
 import unittest