Skip to content

Commit fc08621

Browse files
njriasanfacebook-github-bot
authored andcommitted
Fix distutils failure in Triton Beta testing
Summary: Fixes the distutils issues similar to D73934713 Reviewed By: bottler Differential Revision: D75631611 fbshipit-source-id: 09c354d8cc51ff2c46f4688d7f674370e3f48f1e
1 parent 3f327a5 commit fc08621

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pytorch3d/implicitron/models/implicit_function/voxel_grid.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,6 @@
2121
import warnings
2222
from collections.abc import Mapping
2323
from dataclasses import dataclass, field
24-
25-
from distutils.version import LooseVersion
2624
from typing import Any, Callable, ClassVar, Dict, Iterator, List, Optional, Tuple, Type
2725

2826
import torch
@@ -222,7 +220,8 @@ def change_resolution(
222220
+ "| 'bicubic' | 'linear' | 'area' | 'nearest-exact'"
223221
)
224222

225-
interpolate_has_antialias = LooseVersion(torch.__version__) >= "1.11"
223+
# We assume PyTorch 1.11 and newer.
224+
interpolate_has_antialias = True
226225

227226
if antialias and not interpolate_has_antialias:
228227
warnings.warn("Antialiased interpolation requires PyTorch 1.11+; ignoring")

0 commit comments

Comments
 (0)