Import TelegramMessage only if necessary

.github/CODEOWNERS

@@ -0,0 +1,2 @@
+* @analysiscenter/codeowners
+

.github/workflows/test-install.yml

@@ -16,7 +16,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: [3.9, '3.10', 3.11, 3.12]
+        python-version: [3.9, '3.10', 3.11, 3.12, 3.13]
 
     runs-on: ${{ matrix.os }}
 
@@ -52,7 +52,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: [3.9, '3.10', 3.11, 3.12]
+        python-version: [3.9, '3.10', 3.11, 3.12, 3.13]
 
     runs-on: ${{ matrix.os }}
 
@@ -87,7 +87,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: [3.9, '3.10', 3.11, 3.12]
+        python-version: [3.9, '3.10', 3.11, 3.12, 3.13]
 
     runs-on: ${{ matrix.os }}
 
@@ -151,7 +151,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: [3.9, '3.10', 3.11, 3.12]
+        python-version: [3.9, '3.10', 3.11, 3.12, 3.13]
 
     runs-on: ${{ matrix.os }}
 

batchflow/__init__.py

@@ -23,7 +23,6 @@
 from .sampler import Sampler, ConstantSampler, NumpySampler, HistoSampler, ScipySampler
 from .utils import save_data_to, read_data_from
 from .utils_random import make_rng, make_seed_sequence, spawn_seed_sequence
-from .utils_telegram import TelegramMessage
 from .utils_transforms import Normalizer, Quantizer
 
 
@@ -34,7 +33,7 @@
 pylint_notebook = make_delayed_import(module='.utils_notebook', package=__name__, attribute='pylint_notebook')
 get_available_gpus = make_delayed_import(module='.utils_notebook', package=__name__, attribute='get_available_gpus')
 set_gpus = make_delayed_import(module='.utils_notebook', package=__name__, attribute='set_gpus')
-
+TelegramMessage = make_delayed_import(module='.utils_telegram', package=__name__, attribute='TelegramMessage')
 
 try:
     __version__ = version('batchflow')

batchflow/notifier.py

@@ -19,7 +19,6 @@
 
 from .named_expr import NamedExpression, eval_expr
 from .monitor import ResourceMonitor, MONITOR_ALIASES
-from .utils_telegram import TelegramMessage
 
 
 
@@ -255,6 +254,8 @@ def __init__(self, bar='a', disable=False, frequency=1, monitors=None, graphs=No
         # Prepare Telegram notifications
         self.telegram = telegram
         if self.telegram or initialize_telegram:
+            from .utils_telegram import TelegramMessage
+
             self.telegram_text = TelegramMessage(token=token, chat_id=chat_id, silent=silent)
             self.telegram_media = TelegramMessage(token=token, chat_id=chat_id, silent=silent)
 

batchflow/utils_telegram.py

@@ -1,7 +1,6 @@
 """ A class to send updateable content messages to Telegram. """
 import os
 import json
-import imghdr
 from uuid import uuid4
 from io import BytesIO
 from concurrent.futures import ThreadPoolExecutor
@@ -148,14 +147,17 @@ def delete(self):
     def content_to_dict(content, group=True, **kwargs):
         """ Convert a content (either a path to photo or matplotlib figure) to a telegram-acceptable dictionary. """
         # Convert content to a bytestring
+        from PIL import Image
+
         if isinstance(content, str):
             bytes_ = TelegramMessage.path_to_bytes(content)
         else:
             bytes_ = TelegramMessage.figure_to_bytes(content, **kwargs)
 
         # Create unique id of attachment, pack entities in json
         attach_id = f'attached{uuid4().hex}'
-        attach_fmt = imghdr.what(None, bytes_)
+        image = Image.open(BytesIO(bytes_))
+        attach_fmt = image.format.lower()
 
         media_dict = {'media': f'attach://{attach_id}', 'type': 'photo'}
         media_dict = [media_dict] if group else media_dict

pyproject.toml

@@ -70,6 +70,10 @@ jupyter = [
     "requests"
 ]
 
+telegram = [
+    "pillow>=9.4,<11.0",
+]
+
 other = [
     "urllib3>=1.25"
 ]