read the docs hot fixes

AIStream-Peelout · Oct 8, 2024 · 8db6916 · 8db6916
1 parent ad5ef19
commit 8db6916
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 10 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,13 +1,27 @@
-sphinx
-sphinx-rtd-theme
-sphinx-autodoc-typehints
+shap==0.46.0
+scikit-learn>=1.0.1
 pandas
-numpy
 torch
-matplotlib
-plotly
+tb-nightly
+seaborn
+future
+h5py
+wandb==0.17.2
+google-cloud
 google-cloud-storage
-scikit-learn
-wandb
-shap
+plotly~=5.24.0
+pytz>=2022.1
+setuptools~=69.5.1
+numpy==1.26
+requests
+torchvision>=0.6.0
+mpld3>=0.5
+numba>=0.50
+sphinx
+sphinx-rtd-theme
+sphinx-autodoc-typehints
+sphinx
 einops
+pytorch-tsmixer
+einsum
+jaxtyping
diff --git a/flood_forecast/preprocessing/pytorch_loaders.py b/flood_forecast/preprocessing/pytorch_loaders.py
@@ -32,6 +32,7 @@ def __init__(
     ):
         """
         A data loader that takes a CSV file and properly batches for use in training/eval a PyTorch model
+
         :param file_path: The path to the CSV file you wish to use (GCS compatible) or a Pandas dataframe.
         :param forecast_history: This is the length of the historical time series data you wish to
                                 utilize for forecasting

diff --git a/flood_forecast/transformer_xl/cross_former.py b/flood_forecast/transformer_xl/cross_former.py
@@ -206,7 +206,7 @@ def forward(self, x):
 
 
 class Encoder(nn.Module):
-    """The Encoder of Crossformer."""
+    """The Encoder of Crossformer model."""
 
     def __init__(
         self,