returnn 1.20250405.2748__tar.gz → 1.20250416.134058__tar.gz

{returnn-1.20250405.2748/returnn.egg-info → returnn-1.20250416.134058}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20250405.2748
+Version: 1.20250416.134058
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20250416.134058/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20250416.134058'
+long_version = '1.20250416.134058+git.ead6299'

{returnn-1.20250405.2748 → returnn-1.20250416.134058}/returnn/datasets/basic.py

@@ -864,18 +864,16 @@ class Dataset:
             data = self.get_data(seq_idx, key)
             return data[s0_start:s0_end]
 
-    def get_tag(self, sorted_seq_idx):
+    def get_tag(self, sorted_seq_idx: int) -> str:
         """
-        :param int sorted_seq_idx:
-        :rtype: str
+        :param sorted_seq_idx:
         """
         return "seq-%i" % sorted_seq_idx
 
-    def get_all_tags(self):
+    def get_all_tags(self) -> List[str]:
         """
         :return: list of all seq tags, of the whole dataset, without partition epoch.
           Note that this is not possible with all datasets.
-        :rtype: list[str]
         """
         raise OptionalNotImplementedError(f"{self} get_all_tags not implemented")
 

{returnn-1.20250405.2748 → returnn-1.20250416.134058}/returnn/datasets/postprocessing.py

@@ -243,6 +243,13 @@ class PostprocessingDataset(CachedDataset2):
         assert self._dataset is not None
         return self._dataset.get_total_num_seqs(fast=fast)
 
+    def get_all_tags(self) -> List[str]:
+        """:return: all tags"""
+        if self._map_seq_stream is not None:
+            raise util.OptionalNotImplementedError(f"{self}: get_all_tags not allowed when map_seq_stream is set.")
+        assert self._dataset is not None
+        return self._dataset.get_all_tags()
+
     def supports_sharding(self) -> bool:
         """:return: whether this dataset supports sharding"""
         assert self._dataset is not None

{returnn-1.20250405.2748 → returnn-1.20250416.134058}/returnn/torch/engine.py

@@ -1472,7 +1472,7 @@ def _print_process(
             info += ["%.3f sec/step" % step_duration]
         if start_elapsed is not None:
             info += ["elapsed %s" % hms(start_elapsed)]
-        if complete_frac is not None:
+        if complete_frac not in (-1, None):
             assert 1 >= complete_frac > 0, f"{step} step, {complete_frac} complete_frac"
             assert start_elapsed is not None
             total_time_estimated = start_elapsed / complete_frac

{returnn-1.20250405.2748 → returnn-1.20250416.134058/returnn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20250405.2748
+Version: 1.20250416.134058
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20250405.2748/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20250405.002748'
-long_version = '1.20250405.002748+git.cf50800'