From 0f010713cac89aeb2a8e18ed6d7ef3b601123ac3 Mon Sep 17 00:00:00 2001
From: cadam00 <115782686+cadam00@users.noreply.github.com>
Date: Tue, 22 Oct 2024 20:01:39 +0300
Subject: [PATCH] Update wast_modules.py

---
 openstl/modules/wast_modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/openstl/modules/wast_modules.py b/openstl/modules/wast_modules.py
index ca76b504..14b1d928 100644
--- a/openstl/modules/wast_modules.py
+++ b/openstl/modules/wast_modules.py
@@ -7,7 +7,7 @@
 from timm.models._efficientnet_blocks import SqueezeExcite, InvertedResidual
 
 # version adaptation for PyTorch > 1.7.1
-IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.'))) > (1, 7, 1)
+IS_HIGH_VERSION = tuple(map(lambda x: int(float(x)), torch.__version__.split('+')[0].split('.')[0:3])) > (1, 7, 1)
 if IS_HIGH_VERSION:
     import torch.fft