From fc08621879e57ae17f9663fd1d3775bcea41e3e2 Mon Sep 17 00:00:00 2001 From: Nick Riasanovsky Date: Thu, 29 May 2025 18:18:49 -0700 Subject: [PATCH] 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 --- pytorch3d/implicitron/models/implicit_function/voxel_grid.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pytorch3d/implicitron/models/implicit_function/voxel_grid.py b/pytorch3d/implicitron/models/implicit_function/voxel_grid.py index 9b9ddbab..60d58544 100644 --- a/pytorch3d/implicitron/models/implicit_function/voxel_grid.py +++ b/pytorch3d/implicitron/models/implicit_function/voxel_grid.py @@ -21,8 +21,6 @@ import logging import warnings from collections.abc import Mapping from dataclasses import dataclass, field - -from distutils.version import LooseVersion from typing import Any, Callable, ClassVar, Dict, Iterator, List, Optional, Tuple, Type import torch @@ -222,7 +220,8 @@ class VoxelGridBase(ReplaceableBase, torch.nn.Module): + "| 'bicubic' | 'linear' | 'area' | 'nearest-exact'" ) - interpolate_has_antialias = LooseVersion(torch.__version__) >= "1.11" + # We assume PyTorch 1.11 and newer. + interpolate_has_antialias = True if antialias and not interpolate_has_antialias: warnings.warn("Antialiased interpolation requires PyTorch 1.11+; ignoring")