Skip to content

Commit 4b3c32d

Browse files
committed
Revert changes in tests
1 parent b1f3e52 commit 4b3c32d

2 files changed

Lines changed: 8 additions & 9 deletions

File tree

metatomic-torch/tests/models.cpp

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -132,15 +132,13 @@ TEST_CASE("Models metadata") {
132132
auto options = torch::make_intrusive<ModelEvaluationOptionsHolder>();
133133
options->set_length_unit("nanometer");
134134

135-
auto output1 = torch::make_intrusive<ModelOutputHolder>();
136-
output1->per_atom = false;
137-
options->outputs.insert("output_1", output1);
135+
options->outputs.insert("output_1", torch::make_intrusive<ModelOutputHolder>());
138136

139-
auto output2 = torch::make_intrusive<ModelOutputHolder>();
140-
output2->per_atom = true;
141-
output2->set_quantity("something");
142-
output2->set_unit("something");
143-
options->outputs.insert("output_2", output2);
137+
auto output = torch::make_intrusive<ModelOutputHolder>();
138+
output->per_atom = true;
139+
output->set_quantity("something");
140+
output->set_unit("something");
141+
options->outputs.insert("output_2", output);
144142

145143
const auto* expected = R"({
146144
"class": "ModelEvaluationOptions",
@@ -194,7 +192,7 @@ TEST_CASE("Models metadata") {
194192
);
195193
CHECK(*options->get_selected_atoms().value() == *expected_selection);
196194

197-
auto output = options->outputs.at("foo");
195+
output = options->outputs.at("foo");
198196
CHECK(output->quantity().empty());
199197
CHECK(output->unit().empty());
200198
CHECK(output->per_atom == false);

python/metatomic_torch/pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ changelog = "https://docs.metatensor.org/metatomic/latest/torch/CHANGELOG.html"
3333
### ======================================================================== ###
3434
[build-system]
3535
requires = [
36+
"torch == 2.9",
3637
"setuptools >=77",
3738
"packaging >=23",
3839
"cmake",

0 commit comments

Comments
 (0)