From 48a24bd82711eb8c340444faf3a21a5d77d28ee0 Mon Sep 17 00:00:00 2001
From: Alessandro Pappalardo <1934033+volcacius@users.noreply.github.com>
Date: Fri, 20 Oct 2023 14:45:12 +0100
Subject: [PATCH] Fix (fx): import backport on 2.1 (#732)

---
 src/brevitas/fx/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/brevitas/fx/__init__.py b/src/brevitas/fx/__init__.py
index ab9e77ffe..50e66a159 100644
--- a/src/brevitas/fx/__init__.py
+++ b/src/brevitas/fx/__init__.py
@@ -7,7 +7,7 @@
 from brevitas import torch_version
 
 # This needs to be bumped until https://github.com/pytorch/pytorch/pull/94461 gets merged
-if torch_version < version.parse('2.1'):
+if torch_version < version.parse('2.2'):
     import brevitas.backport.fx as fx
     from brevitas.backport.fx._compatibility import compatibility
     from brevitas.backport.fx._symbolic_trace import _assert_is_none