diff --git a/.gitignore b/.gitignore index e0fad7f..9fc0cb4 100644 --- a/.gitignore +++ b/.gitignore @@ -12,4 +12,5 @@ build/ *.json result* data/cad -misc \ No newline at end of file +misc +.venv/ \ No newline at end of file diff --git a/cpp_to_py/io_parser/gp/GPDatabase.cpp b/cpp_to_py/io_parser/gp/GPDatabase.cpp index 574a03b..5a762f2 100644 --- a/cpp_to_py/io_parser/gp/GPDatabase.cpp +++ b/cpp_to_py/io_parser/gp/GPDatabase.cpp @@ -487,7 +487,7 @@ std::vector GPDatabase::getHyperedgeInfoTensor() { } auto hyperedge_index = torch::cat({hyperedge_list.unsqueeze(0), hyperedge_index_helper.unsqueeze(0)}, 0); - auto new_order_idx = torch::argsort(hyperedge_index.index({0}), 0); + auto new_order_idx = torch::argsort(hyperedge_index.index({0}), 0, false); hyperedge_index = hyperedge_index.index({torch::indexing::Slice(), new_order_idx}); return {hyperedge_index, hyperedge_list, hyperedge_list_end};