logger escape html in notebooks

Author: vpj

client/labml/__init__.py

@@ -1 +1 @@
-__version__ = '0.5.1'
+__version__ = '0.5.2'

client/labml/internal/logger/destinations/ipynb.py

@@ -33,6 +33,13 @@ def is_same_cell(self):
 
         return False
 
+    @staticmethod
+    def _escape_html(text: str) -> str:
+        text = text.replace('&', '&')
+        text = text.replace('<', '&lt;')
+        text = text.replace('>', '&gt;')
+        return text
+
     @staticmethod
     def html_code(text: str, color: List[StyleCode] or StyleCode or None):
         """
@@ -42,6 +49,8 @@ def html_code(text: str, color: List[StyleCode] or StyleCode or None):
             assert color is None
             return text
 
+        text = IpynbDestination._escape_html(text)
+
         if color is None:
             return text
         elif isinstance(color, list):