fix: update python version and fix bug in vis test (TRI-ML#85)

* fix: update python version and fix bug in vis test

* fix: updated workflows

.github/workflows/coverage.yml

@@ -12,7 +12,7 @@ jobs:
     env:
       ENVIRONMENT: TEST_RUNNER
       OS: ubuntu-18.04
-      PYTHON: '3.6'
+      PYTHON: '3.7'
       COVERAGE_TOTAL: 46 # Coverage threshold percentage
     steps:
     - name: Checkout (admin token)
@@ -29,7 +29,7 @@ jobs:
     - name: Setup Python
       uses: actions/setup-python@master
       with:
-        python-version: 3.6
+        python-version: 3.7
     - name: Install dependencies
       id: coverage-installer
       run: |

.github/workflows/doc-gen.yml

@@ -8,15 +8,15 @@ jobs:
     runs-on: ubuntu-18.04
     env:
       OS: ubuntu-18.04
-      PYTHON: '3.6'
+      PYTHON: '3.7'
     steps:
     - uses: actions/checkout@master
       with:
         fetch-depth: ‘1’
     - name: Setup Python
       uses: actions/setup-python@master
       with:
-        python-version: 3.6
+        python-version: 3.7
     - name: Setup requirements and run sphinx
       run: |
         python -m pip install --upgrade pip

.github/workflows/pre-merge.yml

@@ -21,10 +21,10 @@ jobs:
         uses: wagoid/commitlint-github-action@v4
         with:
           configFile: .commitlintrc.yml
-      - name: Install Python 3.6
+      - name: Install Python 3.7
         uses: actions/setup-python@v1
         with:
-          python-version: 3.6
+          python-version: 3.7
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip

Dockerfile

@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.1-devel-ubuntu18.04
 
-ARG python=3.6
+ARG python=3.7
 ENV PYTORCH_VERSION=1.8.1+cu111
 ENV TORCHVISION_VERSION=0.9.1+cu111
 

tests/utils/test_visualization_engine.py

@@ -1,11 +1,13 @@
 import json
 import os
-import cv2
 
+import cv2
 import numpy as np
-from dgp.datasets.synchronized_dataset import SynchronizedScene
-from dgp.utils.visualization_engine import visualize_dataset_3d, visualize_dataset_2d, visualize_dataset_sample_3d, visualize_dataset_sample_2d
 
+from dgp.datasets.synchronized_dataset import SynchronizedScene
+from dgp.utils.visualization_engine import (
+    visualize_dataset_2d, visualize_dataset_3d, visualize_dataset_sample_2d, visualize_dataset_sample_3d
+)
 from tests import TEST_DATA_DIR
 
 
@@ -78,5 +80,5 @@ def test_visualize_dataset_sample_3d():
     )
 
     result = visualize_dataset_sample_3d(dataset=dataset, scene_idx=0, sample_idx=0, camera_datum_names=['camera_05'])
-    data = cv2.imread('tests/data/dgp/vis_output.png', cv2.IMREAD_COLOR)
-    assert np.allclose(result["camera_05"], data)
+    data = cv2.cvtColor(cv2.imread('tests/data/dgp/vis_output.png', cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
+    assert np.allclose(result['camera_05'], data, rtol=1e-3)