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
This commit is contained in:
Nick Riasanovsky 2025-05-29 18:18:49 -07:00 committed by Facebook GitHub Bot
parent 3f327a516b
commit fc08621879

View File

@ -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")