Merge pull request #71 from cms-cat/fixtest

fix: temp style workaround for jupyter

Author: oglez

src/cmsstyle/__init__.py

@@ -1 +1,3 @@
+from ._version import version as __version__  # noqa: F401
+
 from .cmsstyle import *

src/cmsstyle/_utils.py

@@ -0,0 +1,9 @@
+def env_type():
+    try:
+        ipy_str = str(type(get_ipython()))
+        if 'zmqshell' in ipy_str:
+            return 'jupyter'
+        if 'terminal' in ipy_str:
+            return 'ipython'
+    except:
+        return 'terminal'

src/cmsstyle/_version.pyi

@@ -0,0 +1,2 @@
+version: str
+version_tuple: tuple[int, int, int] | tuple[int, int, int, str, str]

src/cmsstyle/cmsstyle.py

@@ -20,6 +20,7 @@
 from dataclasses import dataclass
 from typing import Any, Iterable
 import os
+from ._utils import env_type
 
 # This global variables for the module should not be accessed directy! Use the utilities below.
 cms_lumi = "Run 2, 138 fb^{#minus1}"
@@ -725,13 +726,13 @@ def setCMSStyle(force=rt.kTRUE):
     cmsStyle.SetHatchesSpacing(1.3)
 
     # Some additional parameters we need to set as "style"
-
-    if (
-        float(".".join(re.split("\\.|/", rt.__version__)[0:2])) >= 6.32
-    ):  # Not available before!
-        # This change by O. Gonzalez allows to save inside the canvas the
-        # informnation about the defined colours.
-        rt.TColor.DefinedColors(1)
+    if env_type() != "jupyter":  # Snippet below crashes in jupyter as late as 6.36. Issue reproted to ROOT
+        if (
+            float(".".join(re.split("\\.|/", rt.__version__)[0:2])) >= 6.32
+        ):  # Not available before!
+            # This change by O. Gonzalez allows to save inside the canvas the
+            # informnation about the defined colours.
+            rt.TColor.DefinedColors(1)
 
     # Using the Style.
     cmsStyle.cd()