Skip to content

Commit c719c69

Browse files
committed
Linting
1 parent 55a69e1 commit c719c69

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pvnet/models/multimodal/site_encoders/encoders.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -151,10 +151,12 @@ def __init__(
151151
target_id_dim: The number of unique IDs.
152152
target_key_to_use: The key to use for the target in the attention layer.
153153
input_key_to_use: The key to use for the input in the attention layer.
154-
num_channels: Number of channels in the input data. For single site generation, this will be 1, as there
155-
is not channel dimension, for Sensors, this will probably be higher than that
156-
num_sites_in_inference: Number of sites to use in inference. This is used to determine the
157-
number of sites to use in the attention layer, for a single site, 1 works, while for multiple sites
154+
num_channels: Number of channels in the input data. For single site generation,
155+
this will be 1, as there is not channel dimension, for Sensors,
156+
this will probably be higher than that
157+
num_sites_in_inference: Number of sites to use in inference.
158+
This is used to determine the number of sites to use in the
159+
attention layer, for a single site, 1 works, while for multiple sites
158160
(such as multiple sensors), this would be higher than that
159161
160162
"""

0 commit comments

Comments
 (0)