From c18ee9d40a3b46d63e427aa02ad82100025e8ad6 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 12 Apr 2021 19:09:17 -0700 Subject: [PATCH] lint fixes Summary: Lint after recent changes. Reviewed By: nikhilaravi Differential Revision: D27682328 fbshipit-source-id: 285d159010d886e4e97902995adbdff875fd3c19 --- tests/test_io_obj.py | 7 ++----- tests/test_r2n2.py | 3 +-- tests/test_rasterize_rectangle_images.py | 2 +- tests/test_render_meshes.py | 2 +- tests/test_render_meshes_clipped.py | 2 +- tests/test_render_points.py | 2 +- tests/test_sample_points_from_meshes.py | 2 +- tests/test_shapenet_core.py | 2 +- 8 files changed, 9 insertions(+), 13 deletions(-) diff --git a/tests/test_io_obj.py b/tests/test_io_obj.py index d5b331de..9d4ca6c3 100644 --- a/tests/test_io_obj.py +++ b/tests/test_io_obj.py @@ -8,11 +8,7 @@ from pathlib import Path from tempfile import NamedTemporaryFile import torch -from common_testing import ( - TestCaseMixin, - get_pytorch3d_dir, - get_tests_dir, -) +from common_testing import TestCaseMixin, get_pytorch3d_dir, get_tests_dir from iopath.common.file_io import PathManager from pytorch3d.io import IO, load_obj, load_objs_as_meshes, save_obj from pytorch3d.io.mtl_io import ( @@ -24,6 +20,7 @@ from pytorch3d.renderer import TexturesAtlas, TexturesUV, TexturesVertex from pytorch3d.structures import Meshes, join_meshes_as_batch from pytorch3d.utils import torus + DATA_DIR = get_tests_dir() / "data" TUTORIAL_DATA_DIR = get_pytorch3d_dir() / "docs/tutorials/data" diff --git a/tests/test_r2n2.py b/tests/test_r2n2.py index 8fcdafab..a4e2bccc 100644 --- a/tests/test_r2n2.py +++ b/tests/test_r2n2.py @@ -5,11 +5,10 @@ Sanity checks for loading R2N2. import json import os import unittest -from pathlib import Path import numpy as np import torch -from common_testing import TestCaseMixin, load_rgb_image, get_tests_dir +from common_testing import TestCaseMixin, get_tests_dir, load_rgb_image from PIL import Image from pytorch3d.datasets import ( R2N2, diff --git a/tests/test_rasterize_rectangle_images.py b/tests/test_rasterize_rectangle_images.py index 4c9e4cae..69816e81 100644 --- a/tests/test_rasterize_rectangle_images.py +++ b/tests/test_rasterize_rectangle_images.py @@ -7,9 +7,9 @@ import numpy as np import torch from common_testing import ( TestCaseMixin, - load_rgb_image, get_pytorch3d_dir, get_tests_dir, + load_rgb_image, ) from PIL import Image from pytorch3d.io import load_obj diff --git a/tests/test_render_meshes.py b/tests/test_render_meshes.py index c585ecd1..d737529e 100644 --- a/tests/test_render_meshes.py +++ b/tests/test_render_meshes.py @@ -11,9 +11,9 @@ import numpy as np import torch from common_testing import ( TestCaseMixin, - load_rgb_image, get_pytorch3d_dir, get_tests_dir, + load_rgb_image, ) from PIL import Image from pytorch3d.io import load_obj diff --git a/tests/test_render_meshes_clipped.py b/tests/test_render_meshes_clipped.py index de03603a..845da8e1 100644 --- a/tests/test_render_meshes_clipped.py +++ b/tests/test_render_meshes_clipped.py @@ -13,7 +13,7 @@ import unittest import imageio import numpy as np import torch -from common_testing import TestCaseMixin, load_rgb_image, get_tests_dir +from common_testing import TestCaseMixin, get_tests_dir, load_rgb_image from pytorch3d.io import save_obj from pytorch3d.renderer.cameras import FoVPerspectiveCameras, look_at_view_transform from pytorch3d.renderer.lighting import PointLights diff --git a/tests/test_render_points.py b/tests/test_render_points.py index cf45e84a..faa99cac 100644 --- a/tests/test_render_points.py +++ b/tests/test_render_points.py @@ -12,9 +12,9 @@ import numpy as np import torch from common_testing import ( TestCaseMixin, - load_rgb_image, get_pytorch3d_dir, get_tests_dir, + load_rgb_image, ) from PIL import Image from pytorch3d.renderer.cameras import ( diff --git a/tests/test_sample_points_from_meshes.py b/tests/test_sample_points_from_meshes.py index 40cf9c3b..714aab10 100644 --- a/tests/test_sample_points_from_meshes.py +++ b/tests/test_sample_points_from_meshes.py @@ -7,8 +7,8 @@ import numpy as np import torch from common_testing import ( TestCaseMixin, - get_random_cuda_device, get_pytorch3d_dir, + get_random_cuda_device, get_tests_dir, ) from PIL import Image diff --git a/tests/test_shapenet_core.py b/tests/test_shapenet_core.py index 99ae6c6e..0268459f 100644 --- a/tests/test_shapenet_core.py +++ b/tests/test_shapenet_core.py @@ -7,7 +7,7 @@ import unittest import numpy as np import torch -from common_testing import TestCaseMixin, load_rgb_image, get_tests_dir +from common_testing import TestCaseMixin, get_tests_dir, load_rgb_image from PIL import Image from pytorch3d.datasets import ShapeNetCore, collate_batched_meshes from pytorch3d.renderer import (