PVNet 3.0.2__tar.gz → 3.0.3__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {PVNet-3.0.2 → PVNet-3.0.3}/PKG-INFO +1 -1
- {PVNet-3.0.2 → PVNet-3.0.3}/PVNet.egg-info/PKG-INFO +1 -1
- PVNet-3.0.3/pvnet/__init__.py +2 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/site_encoders/encoders.py +2 -0
- PVNet-3.0.2/pvnet/__init__.py +0 -2
- {PVNet-3.0.2 → PVNet-3.0.3}/LICENSE +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/MANIFEST.in +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/PVNet.egg-info/SOURCES.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/PVNet.egg-info/dependency_links.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/PVNet.egg-info/requires.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/PVNet.egg-info/top_level.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/README.md +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/callbacks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/base.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/datamodule.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/pv_site_datamodule.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/utils.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/data/wind_datamodule.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/base_model.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/baseline/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/baseline/last_value.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/baseline/single_value.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/basic_blocks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/deep_supervision.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/encoders/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/encoders/basic_blocks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/encoders/encoders2d.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/encoders/encoders3d.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/encoders/encodersRNN.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/linear_networks/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/linear_networks/basic_blocks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/linear_networks/networks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/multimodal.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/nwp_weighting.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/site_encoders/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/site_encoders/basic_blocks.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/multimodal/weather_residual.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/models/utils.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/optimizers.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/training.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pvnet/utils.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/pyproject.toml +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/requirements-dev.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/requirements.txt +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/setup.cfg +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/setup.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/tests/__init__.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/tests/conftest.py +0 -0
- {PVNet-3.0.2 → PVNet-3.0.3}/tests/test_end2end.py +0 -0
|
@@ -200,6 +200,8 @@ class SingleAttentionNetwork(AbstractPVSitesEncoder):
|
|
|
200
200
|
else:
|
|
201
201
|
ids = x[BatchKey[f"{self.input_key_to_use}_id"]][:, 0]
|
|
202
202
|
ids = ids.squeeze().int()
|
|
203
|
+
if len(ids.shape) == 2: # Batch was squeezed down to nothing
|
|
204
|
+
ids = ids.unsqueeze(0)
|
|
203
205
|
query = self.target_id_embedding(ids).unsqueeze(1)
|
|
204
206
|
return query
|
|
205
207
|
|
PVNet-3.0.2/pvnet/__init__.py
DELETED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|