From 070fd8149ab368950e1d87cf8e72f637012669b6 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 9 Apr 2020 11:35:51 +0300
Subject: [PATCH] Store the version that matching the Session API so we do not
 reload every time

---
 trains/backend_api/api_proxy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trains/backend_api/api_proxy.py b/trains/backend_api/api_proxy.py
index 5d7888fc..0b5ef44e 100644
--- a/trains/backend_api/api_proxy.py
+++ b/trains/backend_api/api_proxy.py
@@ -31,7 +31,7 @@ class ApiServiceProxy(object):
                         ]])
 
             version = str(min(Version(Session.api_version), ApiServiceProxy._max_available_version))
-            self.__dict__["__wrapped_version__"] = version
+            self.__dict__["__wrapped_version__"] = Session.api_version
             name = ".v{}.{}".format(
                 version.replace(".", "_"), self.__dict__.get("__wrapped_name__")
             )