Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 21, 2024
1 parent 9cceef4 commit f748d58
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -61,4 +61,4 @@ pyamg>=5.0.0
git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
onnx_graphsurgeon
polygraphy
e3nn
e3nn
3 changes: 1 addition & 2 deletions tests/test_eq_feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@
import torch
from e3nn import o3
from e3nn.nn import SO3Activation
from tests.utils import testing_data_config, download_url_or_skip_test

def s2_near_identity_grid(max_beta: float = math.pi / 8, n_alpha: int = 8, n_beta: int = 3) -> torch.Tensor:
beta = torch.arange(1, n_beta + 1) * max_beta / n_beta
Expand Down Expand Up @@ -209,7 +208,7 @@ def main():
with torch.no_grad():
features = model(input_slice)

print("Extracted features:", features) #print out extracted features from the equivariant filter
print("Extracted features:", features) #print out extracted features from the equivariant filter

#Save features as .nii.gz files
#save_features_as_nii(features, output_dir="nii_features")
Expand Down

0 comments on commit f748d58

Please sign in to comment.