timelined_array 0.1.1__tar.gz → 0.1.2__tar.gz

{timelined_array-0.1.1 → timelined_array-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: timelined_array
-Version: 0.1.1
+Version: 0.1.2
 Summary: Manage easily 1 or multidimensionnal samples numpy arrays that are time related. Extends numpy without removing any of it's abilities on such arrays.
 Author-Email: Timothe Jost <44769559+JostTim@users.noreply.github.com>
 License: MIT
@@ -16,7 +16,7 @@ Description-Content-Type: text/markdown
 ![PyPI - Version](https://img.shields.io/pypi/v/timelined_array)
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JostTim/timelined_array/publish.yml?label=Publishing)
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JostTim/timelined_array/test.yml?label=Testing)
-![Codecov](https://img.shields.io/codecov/c/github/JostTim/timelined_array)
+[![Codecov](https://img.shields.io/codecov/c/github/JostTim/timelined_array)](https://codecov.io/gh/JostTim/timelined_array)
 
 
 <!-- eventually, add ?cache-bust=1 to badges url above, and update the value of cache-bust randomly at git commits (or based on the commit name) so that we always see the latest badge produced, and not the cached badge in the browser.-->

{timelined_array-0.1.1 → timelined_array-0.1.2}/README.md

@@ -4,7 +4,7 @@
 ![PyPI - Version](https://img.shields.io/pypi/v/timelined_array)
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JostTim/timelined_array/publish.yml?label=Publishing)
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JostTim/timelined_array/test.yml?label=Testing)
-![Codecov](https://img.shields.io/codecov/c/github/JostTim/timelined_array)
+[![Codecov](https://img.shields.io/codecov/c/github/JostTim/timelined_array)](https://codecov.io/gh/JostTim/timelined_array)
 
 
 <!-- eventually, add ?cache-bust=1 to badges url above, and update the value of cache-bust randomly at git commits (or based on the commit name) so that we always see the latest badge produced, and not the cached badge in the browser.-->

{timelined_array-0.1.1 → timelined_array-0.1.2}/pyproject.toml

@@ -10,7 +10,7 @@ dependencies = [
 requires-python = ">=3.11"
 readme = "README.md"
 dynamic = []
-version = "0.1.1"
+version = "0.1.2"
 
 [project.license]
 text = "MIT"

{timelined_array-0.1.1 → timelined_array-0.1.2}/src/timelined_array/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.1.1"
+__version__ = "0.1.2"
 
 from .time import TimelinedArray, MaskedTimelinedArray, Timeline

{timelined_array-0.1.1 → timelined_array-0.1.2}/src/timelined_array/time.py

@@ -7,7 +7,9 @@ import operator
 
 from numpy.typing import NDArray
 from types import MethodType
-from typing import Tuple, List, Protocol, Type, Callable, Any, Optional
+from typing import Tuple, List, Protocol, Type, Callable, Any, Optional, TypeVar, Sequence
+
+Tp = TypeVar("Tp", bound="TimelinedArray")
 
 OperatorType = Callable[[Any, Any], bool]
 
@@ -635,8 +637,10 @@ class BaseTimeArray(TimeMixin, np.ndarray):
         Returns:
             TimelinedArray: The synchronized and cut arrray.
         """
+        # this slice is to select the number of elements, on the time_dimension
+        slices = tuple(slice(None) if i != self.time_dimension else slice(None, element_nb) for i in range(self.ndim))
 
-        return self.itime[start:][:element_nb]  # type: ignore
+        return self.itime[start:][slices]  # type: ignore
 
     def shift_values(
         self,
@@ -829,7 +833,7 @@ class BaseTimeArray(TimeMixin, np.ndarray):
     def all_axes(self):
         return tuple([i for i in range(self.ndim)])
 
-    def mean(self, axis: int | Tuple[int, ...] | None = None, dtype=None, out=None, keepdims=False):
+    def mean(self: Tp, axis: int | Tuple[int, ...] | None = None, dtype=None, out=None, keepdims=False) -> Tp:
         """Calculates the mean along the specified axis.
 
         Args:
@@ -1238,7 +1242,7 @@ class TimelinedArray(BaseTimeArray):
         return type(self).__name__ + np.asarray(self).__str__()
 
     @staticmethod
-    def align_from_iterable(iterable) -> "TimelinedArray":
+    def align_from_iterable(iterable: "Sequence[TimelinedArray]") -> "TimelinedArray":
         """Aligns arrays from an iterable based on their timelines.
 
         Args: