From e96f6df2ac66906722fc6e9b8714d50e0bb9c3c4 Mon Sep 17 00:00:00 2001 From: Douwe den Blanken Date: Tue, 28 Mar 2023 14:28:54 +0200 Subject: [PATCH] Fix issue "UserWarning: Positional args are being deprecated, use kwargs instead. Refer to https://pytorch.org/docs/master/generated/torch.nn.Module.html#torch.nn.Module.state_dict for details." --- src/brevitas/quant_tensor/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/brevitas/quant_tensor/__init__.py b/src/brevitas/quant_tensor/__init__.py index a95b0ffdf..b1907589c 100644 --- a/src/brevitas/quant_tensor/__init__.py +++ b/src/brevitas/quant_tensor/__init__.py @@ -79,6 +79,7 @@ def training(self): else: return None + @classmethod def __torch_function__(self, func, types, args=(), kwargs=None): if kwargs is None: kwargs = {}