lt-tensor 0.0.1.dev1__tar.gz → 0.0.1.dev3__tar.gz

{lt_tensor-0.0.1.dev1 → lt_tensor-0.0.1.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1.dev1
+Version: 0.0.1.dev3
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336

lt_tensor-0.0.1.dev3/lt_tensor/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.1dev3"

{lt_tensor-0.0.1.dev1 → lt_tensor-0.0.1.dev3}/lt_tensor/misc_utils.py

@@ -23,7 +23,7 @@ def log_tensor(
         _b = 20 + len(title.strip())
     print(f"shape: {item.shape}")
     print(f"dtype: {item.dtype}")
-    if not print_details:
+    if print_details:
         print(f"ndim: {item.ndim}")
         if isinstance(item, Tensor):
             print(f"device: {item.device}")

{lt_tensor-0.0.1.dev1 → lt_tensor-0.0.1.dev3}/lt_tensor/transform.py

@@ -175,7 +175,7 @@ def get_sinusoidal_embedding(timesteps: torch.Tensor, dim: int) -> torch.Tensor:
     return emb
 
 
-def generate_window(
+def _generate_window(
     M: int, alpha: float = 0.5, device: Optional[DeviceType] = None
 ) -> Tensor:
     if M < 1:
@@ -260,7 +260,7 @@ def window_sumsquare(
     x = torch.zeros(total_length, dtype=dtype, device=device)
 
     # Get the window (from scipy for now)
-    win = generate_window(window_spec, win_length, fftbins=True)
+    win = _generate_window(window_spec, win_length, fftbins=True)
     win = torch.tensor(win, dtype=dtype, device=device)
 
     # Normalize and square
@@ -277,9 +277,6 @@ def window_sumsquare(
     return x
 
 
-def get_window(win_length: int = 1200):
-    return generate_window(win_length)
-
 
 def inverse_transform(
     spec: Tensor,
@@ -288,10 +285,10 @@ def inverse_transform(
     n_fft: int = 2048,
     hop_length: int = 300,
     win_length: int = 1200,
-    length: Optional[torch.shape] = None,
+    length: Optional[Any] = None,
 ):
     if window is None:
-        window = generate_window(win_length)
+        window = _generate_window(win_length)
     return torch.istft(
         spec * torch.exp(phase * 1j),
         n_fft,
@@ -313,7 +310,7 @@ def stft_istft_rebuild(
     Perform STFT followed by ISTFT reconstruction using magnitude and phase.
     """
     if window is None:
-        window = generate_window(win_length)
+        window = _generate_window(win_length)
     st = torch.stft(
         input_data,
         n_fft,

{lt_tensor-0.0.1.dev1 → lt_tensor-0.0.1.dev3}/lt_tensor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lt-tensor
-Version: 0.0.1.dev1
+Version: 0.0.1.dev3
 Summary: General utilities for PyTorch and others. Built for general use.
 Home-page: https://github.com/gr1336/lt-tensor/
 Author: gr1336

{lt_tensor-0.0.1.dev1 → lt_tensor-0.0.1.dev3}/setup.py

@@ -4,7 +4,7 @@ with open("README.md", "r", encoding="utf-8") as f:
     long_description = f.read()
 
 setup(
-    version="0.0.1dev1",
+    version="0.0.1dev3",
     name="lt-tensor",
     description="General utilities for PyTorch and others. Built for general use.",
     long_description=long_description,

lt_tensor-0.0.1.dev1/lt_tensor/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.0.1dev1"