dgenerate-ultralytics-headless 8.3.190__py3-none-any.whl → 8.3.192__py3-none-any.whl

ultralytics/utils/tqdm.py

@@ -165,11 +165,18 @@ class TQDM:
 
         # For bytes with scaling, use binary units
         if self.unit in ("B", "bytes") and self.unit_scale:
-            for threshold, unit in [(1073741824, "GB/s"), (1048576, "MB/s"), (1024, "KB/s")]:  # 1 << 30, << 20, << 10
-                if rate >= threshold:
-                    return f"{rate / threshold:.1f}{unit}"
-            return f"{rate:.1f}B/s"
-
+            return next(
+                (
+                    f"{rate / threshold:.1f}{unit}"
+                    for threshold, unit in [
+                        (1073741824, "GB/s"),
+                        (1048576, "MB/s"),
+                        (1024, "KB/s"),
+                    ]
+                    if rate >= threshold
+                ),
+                f"{rate:.1f}B/s",
+            )
         # For other scalable units, use decimal units
         if self.unit_scale and self.unit in ("it", "items", ""):
             for threshold, prefix in [(1000000, "M"), (1000, "K")]:
@@ -210,7 +217,7 @@ class TQDM:
         filled = int(frac * width)
         bar = "━" * filled + "─" * (width - filled)
         if filled < width and frac * width - filled > 0.5:
-            bar = bar[:filled] + "╸" + bar[filled + 1 :]
+            bar = f"{bar[:filled]}╸{bar[filled + 1 :]}"
         return bar
 
     def _should_update(self, dt: float, dn: int) -> bool:
@@ -218,10 +225,7 @@ class TQDM:
         if self.noninteractive:
             return False
 
-        if self.total and self.n >= self.total:
-            return True
-
-        return dt >= self.mininterval
+        return True if self.total and self.n >= self.total else dt >= self.mininterval
 
     def _display(self, final: bool = False) -> None:
         """Display progress bar."""
@@ -258,8 +262,9 @@ class TQDM:
 
         # Calculate remaining time
         remaining_str = ""
-        if self.total and 0 < self.n < self.total and rate > 0:
-            remaining_str = self._format_time((self.total - self.n) / rate)
+        if self.total and 0 < self.n < self.total and elapsed > 0:
+            est_rate = rate or self.n / elapsed
+            remaining_str = self._format_time((self.total - self.n) / est_rate)
 
         # Build progress components
         if self.total:
@@ -449,8 +454,7 @@ if __name__ == "__main__":
 
     def process_files():
         """Simulate processing files of unknown count."""
-        files = [f"file_{i}.txt" for i in range(18)]
-        return files
+        return [f"file_{i}.txt" for i in range(18)]
 
     pbar = TQDM(desc="Scanning files", unit="files")
     files = process_files()

ultralytics/utils/triton.py

@@ -1,6 +1,7 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from typing import List
+from __future__ import annotations
+
 from urllib.parse import urlsplit
 
 import numpy as np
@@ -86,7 +87,7 @@ class TritonRemoteModel:
         self.output_names = [x["name"] for x in config["output"]]
         self.metadata = eval(config.get("parameters", {}).get("metadata", {}).get("string_value", "None"))
 
-    def __call__(self, *inputs: np.ndarray) -> List[np.ndarray]:
+    def __call__(self, *inputs: np.ndarray) -> list[np.ndarray]:
         """
         Call the model with the given inputs and return inference results.