def __init__()

in utils/renderer.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, face_topo_file: str):
        """
        :param face_topo_file: .obj file containing face topology
        """
        if th.cuda.is_available():
            self.device = th.device("cuda:0")
            th.cuda.set_device(self.device)
        else:
            self.device = th.device("cpu")
        verts, faces_idx, _ = load_obj(face_topo_file)
        self.verts_rgb = th.ones_like(verts)[None] * th.Tensor([0.529, 0.807, 0.980])[None, None, :]  # (1, V, 3)
        self.verts_rgb = self.verts_rgb.to(self.device)
        self.faces = faces_idx.verts_idx.to(self.device)[None, :, :]