Add TID252, bans relative imports except siblings

infinigen/assets/materials/art.py

@@ -6,12 +6,12 @@
 from numpy.random import uniform
 
 from infinigen.assets.materials.fabrics.fabric_random import fabric_shader_list
+from infinigen.assets.utils.decorate import read_uv, write_uv
+from infinigen.core.nodes import Nodes, NodeWrangler
 from infinigen.core.util.math import FixedSeed
 from infinigen.core.util.random import log_uniform
 from infinigen.core.util.random import random_general as rg
 
-from ...core.nodes import Nodes, NodeWrangler
-from ..utils.decorate import read_uv, write_uv
 from . import rug, text
 
 

infinigen/assets/materials/fabrics/fabric_random.py

@@ -3,10 +3,10 @@
 # This source code is licensed under the BSD 3-Clause license found in the LICENSE file in the root directory
 # of this source tree.
 
+from infinigen.assets.materials import common
+from infinigen.assets.utils.uv import unwrap_faces
 from infinigen.core.util.random import random_general as rg
 
-from ...utils.uv import unwrap_faces
-from .. import common
 from .coarse_knit_fabric import shader_coarse_knit_fabric
 from .fine_knit_fabric import shader_fine_knit_fabric
 from .leather import shader_leather

infinigen/assets/materials/hardwood_floor.py

@@ -6,9 +6,9 @@
 from numpy.random import uniform
 
 from infinigen.assets.utils.object import new_plane
+from infinigen.core.nodes import Nodes, NodeWrangler
+from infinigen.core.util.random import log_uniform
 
-from ...core.nodes import Nodes, NodeWrangler
-from ...core.util.random import log_uniform
 from . import common
 from .table_materials import shader_wood
 from .utils.surface_utils import perturb_coordinates

infinigen/assets/materials/metal/__init__.py

@@ -7,12 +7,12 @@
 import numpy as np
 from numpy.random import uniform
 
+from infinigen.assets.materials import common
+from infinigen.assets.materials.bark_random import hex_to_rgb
 from infinigen.core.util.color import hsv2rgba, rgb2hsv
 from infinigen.core.util.random import log_uniform
 from infinigen.core.util.random import random_general as rg
 
-from .. import common
-from ..bark_random import hex_to_rgb
 from . import (
     brushed_metal,
     galvanized_metal,

infinigen/assets/materials/tile.py

@@ -9,10 +9,11 @@
 import numpy as np
 from numpy.random import uniform
 
-from ...core.nodes import Nodes, NodeWrangler
-from ...core.util.math import FixedSeed
-from ...core.util.random import log_uniform
-from ..utils.object import new_cube
+from infinigen.assets.utils.object import new_cube
+from infinigen.core.nodes import Nodes, NodeWrangler
+from infinigen.core.util.math import FixedSeed
+from infinigen.core.util.random import log_uniform
+
 from . import ceramic, common
 from .utils.surface_utils import perturb_coordinates
 

infinigen/assets/materials/woods/crossed_wood_tile.py

@@ -3,7 +3,8 @@
 
 
 # Authors: Lingjie Mei
-from .. import tile
+from infinigen.assets.materials import tile
+
 from .wood import shader_wood
 
 

infinigen/assets/materials/woods/hexagon_wood_tile.py

@@ -3,7 +3,8 @@
 
 
 # Authors: Lingjie Mei
-from .. import tile
+from infinigen.assets.materials import tile
+
 from .wood import shader_wood
 
 

infinigen/assets/materials/woods/non_wood_tile.py

@@ -15,7 +15,8 @@ def apply(
     shape=None,
     **kwargs,
 ):
-    from .. import tile
+    from infinigen.assets.materials import tile
+
     from .wood import shader_wood
 
     shader_funcs = tile.get_shader_funcs()

infinigen/assets/materials/woods/square_wood_tile.py

@@ -3,7 +3,8 @@
 
 
 # Authors: Lingjie Mei
-from .. import tile
+from infinigen.assets.materials import tile
+
 from .wood import shader_wood
 
 

infinigen/assets/materials/woods/staggered_wood_tile.py

@@ -3,7 +3,8 @@
 
 
 # Authors: Lingjie Mei
-from .. import tile
+from infinigen.assets.materials import tile
+
 from .wood import shader_wood
 
 

infinigen/assets/materials/woods/wood.py

@@ -8,12 +8,11 @@
 from numpy.random import uniform
 
 from infinigen.assets.materials import common
+from infinigen.assets.utils.object import new_cube
 from infinigen.core.nodes import Nodes, NodeWrangler
 from infinigen.core.util.color import hsv2rgba, rgb2hsv
 from infinigen.core.util.random import log_uniform
 
-from ...utils.object import new_cube
-
 
 def get_color():
     from infinigen.assets.materials.bark_random import get_random_bark_params

infinigen/assets/objects/seating/bed.py

@@ -45,7 +45,7 @@ def mattress_factory(self):
 
     @cached_property
     def quilt_factory(self):
-        from ..clothes.blanket import BlanketFactory
+        from infinigen.assets.objects.clothes.blanket import BlanketFactory
 
         factory = BlanketFactory(self.factory_seed, self.coarse)
         factory.width = self.mattress_factory.width * uniform(1.4, 1.6)
@@ -54,7 +54,7 @@ def quilt_factory(self):
 
     @cached_property
     def comforter_factory(self):
-        from ..clothes.blanket import ComforterFactory
+        from infinigen.assets.objects.clothes.blanket import ComforterFactory
 
         factory = ComforterFactory(self.factory_seed, self.coarse)
         factory.width = self.mattress_factory.width * uniform(1.4, 1.8)
@@ -63,7 +63,7 @@ def comforter_factory(self):
 
     @cached_property
     def box_comforter_factory(self):
-        from ..clothes.blanket import BoxComforterFactory
+        from infinigen.assets.objects.clothes.blanket import BoxComforterFactory
 
         factory = BoxComforterFactory(self.factory_seed, self.coarse)
         factory.width = self.mattress_factory.width * uniform(1.4, 1.8)
@@ -72,7 +72,7 @@ def box_comforter_factory(self):
 
     @cached_property
     def cover_factory(self):
-        from ..clothes.blanket import BlanketFactory
+        from infinigen.assets.objects.clothes.blanket import BlanketFactory
 
         factory = BlanketFactory(self.factory_seed, self.coarse)
         factory.width = self.mattress_factory.width * uniform(1.6, 1.8)
@@ -81,7 +81,7 @@ def cover_factory(self):
 
     @cached_property
     def towel_factory(self):
-        from ..clothes import TowelFactory
+        from infinigen.assets.objects.clothes import TowelFactory
 
         return TowelFactory(self.factory_seed)
 

infinigen/assets/scatters/ground_twigs.py

@@ -8,10 +8,10 @@
 from numpy.random import uniform as U
 
 from infinigen.assets.objects.trees.generate import make_twig_collection, random_species
+from infinigen.assets.utils.misc import toggle_hide, toggle_show
 from infinigen.core import surface
 from infinigen.core.placement.instance_scatter import scatter_instances
 
-from ..utils.misc import toggle_hide, toggle_show
 from .chopped_trees import approx_settle_transform
 
 

infinigen/core/constraints/example_solver/moves/pose.py

@@ -9,10 +9,10 @@
 
 import numpy as np
 
+from example_solver.state_def import State
 from infinigen.core.constraints.constraint_language import util as iu
 from infinigen.core.constraints.example_solver.geometry import dof, validity
 
-from ..state_def import State
 from . import moves
 from .reassignment import pose_backup, restore_pose_backup
 

infinigen/core/constraints/example_solver/moves/reassignment.py

@@ -9,10 +9,10 @@
 import copy
 from dataclasses import dataclass
 
+from example_solver.state_def import ObjectState, State
 from infinigen.core.constraints.constraint_language.util import sync_trimesh
 from infinigen.core.constraints.example_solver.geometry import dof
 
-from ..state_def import ObjectState, State
 from . import moves
 
 

infinigen/core/constraints/reasoning/expr_equal.py

@@ -6,7 +6,7 @@
 
 import dataclasses
 
-from ..constraint_language.types import Node
+from infinigen.core.constraints.constraint_language.types import Node
 
 
 @dataclasses.dataclass

infinigen/core/nodes/node_transpiler/transpiler.py

@@ -16,7 +16,7 @@
 import mathutils
 import numpy as np
 
-from ..node_info import OUTPUT_NODE_IDS, SINGLETON_NODES, Nodes
+from infinigen.core.nodes.node_info import OUTPUT_NODE_IDS, SINGLETON_NODES, Nodes
 
 logger = logging.getLogger(__name__)
 

infinigen/core/placement/factory.py

@@ -14,10 +14,10 @@
 import numpy as np
 from tqdm import trange
 
+from infinigen.assets.utils.object import center
 from infinigen.core.util import blender as butil
 from infinigen.core.util.math import FixedSeed, int_hash
 
-from ...assets.utils.object import center
 from . import detail
 
 logger = logging.getLogger(__name__)

infinigen/tools/ground_truth/bounding_boxes_3d.py

@@ -22,8 +22,8 @@
         "GT visualization requires `einops`. Please install optional extras via `pip install .[vis]`."
     )
 
-from ..compress_masks import recover
-from ..dataset_loader import get_frame_path
+from infinigen.tools.compress_masks import recover
+from infinigen.tools.dataset_loader import get_frame_path
 
 """
 Usage: python -m tools.ground_truth.bounding_boxes_3d <scene-folder> <frame-index> [--query <query>]

infinigen/tools/ground_truth/depth_to_normals.py

@@ -13,7 +13,7 @@
 from imageio.v3 import imread, imwrite
 from numpy.linalg import inv
 
-from ..dataset_loader import get_frame_path
+from infinigen.tools.dataset_loader import get_frame_path
 
 try:
     from einops import einsum

infinigen/tools/ground_truth/optical_flow_warp.py

@@ -10,7 +10,7 @@
 import numpy as np
 from imageio.v3 import imread, imwrite
 
-from ..dataset_loader import get_frame_path
+from infinigen.tools.dataset_loader import get_frame_path
 
 """
 Usage: python -m tools.ground_truth.rigid_warp <scene-folder> <frame-index-i>

infinigen/tools/ground_truth/rigid_warp.py

@@ -11,7 +11,7 @@
 from imageio.v3 import imread, imwrite
 from numpy.linalg import inv
 
-from ..dataset_loader import get_frame_path
+from infinigen.tools.dataset_loader import get_frame_path
 
 try:
     from einops import einsum

infinigen/tools/ground_truth/segmentation_lookup.py

@@ -15,8 +15,8 @@
 from imageio.v3 import imread, imwrite
 from numba.types import bool_
 
-from ..compress_masks import recover
-from ..dataset_loader import get_frame_path
+from infinigen.tools.compress_masks import recover
+from infinigen.tools.dataset_loader import get_frame_path
 
 try:
     from einops import pack, rearrange, repeat

pyproject.toml

@@ -145,7 +145,8 @@ select = [
     "E", 
     "I", 
     "F",
-    "S102", # no exec()
+    "S102", # no exec(),
+    "TID252", # relative imports only allowed for siblings
 
 ]
 ignore = [