amend

pytorch · Feb 13, 2024 · 153325c · 153325c
1 parent 97dc726
commit 153325c
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tensordict/nn/functional_modules.py b/tensordict/nn/functional_modules.py
@@ -6,19 +6,14 @@
 from __future__ import annotations
 
 import inspect
-import os
 import re
 import types
 import warnings
 from copy import deepcopy
-from distutils.util import strtobool
 from functools import wraps
 from typing import Any, Callable, Iterable
 
-import tensordict.nn.utils
-
 import torch
-from tensordict._contextlib import _DecoratorContextManager
 from tensordict._pytree import PYTREE_REGISTERED_TDS
 
 from tensordict._td import TensorDict