From d0c040dc25c47124b40e600f81d0455f408c864b Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 7 Dec 2023 16:07:16 +0200
Subject: [PATCH] Better message on "Enter"

---
 clearml_session/__main__.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/clearml_session/__main__.py b/clearml_session/__main__.py
index 0c83715..ca08722 100644
--- a/clearml_session/__main__.py
+++ b/clearml_session/__main__.py
@@ -15,6 +15,17 @@ from time import time, sleep
 if sys.platform == 'win32':
     import msvcrt  # noqa
     import wexpect as pexpect  # noqa
+
+    if sys.prefix != sys.base_prefix:
+        # running in venv
+        from clearml.utilities.version import Version
+        if Version(pexpect.__version__) < Version("4.0.1"):
+            # Wexpect issue https://github.com/raczben/wexpect/issues/26#issuecomment-1574339194
+            try:
+                import wexpect_venv as pexpect
+            except ImportError:
+                print("WARNING: wexpect_venv is required to connect via SSH when running in a Windows virtualenv."
+                      " Please install using 'pip install wexpect_venv' in this virtual environment")
 else:
     import select  # noqa
     import pexpect  # noqa