Skip to content

Commit

Permalink
try to fix ci errors
Browse files Browse the repository at this point in the history
  • Loading branch information
xbw886 committed Dec 6, 2024
1 parent 49111f2 commit e30437e
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
3 changes: 1 addition & 2 deletions yacl/examples/sse/sse.cc
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,7 @@ Sse::Sse(int bucket_size, int slot_size, int lambda, int n_lambda)
std::string Sse::GetKt() { return k_map_["Kt"]; }

// EDBSetup
std::pair<std::vector<std::vector<TSet::Record>>, std::string>
Sse::EDBSetup() {
std::pair<std::vector<std::vector<TSet::Record>>, std::string> Sse::EDBSetup() {
ProcessAndUpdateTAndXSet();
auto [TSet, Kt] = tset_.TSetSetup(T_, keywords_);
TSet_ = TSet;
Expand Down
5 changes: 2 additions & 3 deletions yacl/examples/sse/sse.h
Original file line number Diff line number Diff line change
Expand Up @@ -86,9 +86,8 @@ class Sse {

void SaveKeys(const std::map<std::string, std::string>& K_map,
const std::string& file_path);
void SaveTSet(
const std::vector<std::vector<TSet::Record>>& TSet,
const std::string& file_path);
void SaveTSet(const std::vector<std::vector<TSet::Record>>& TSet,
const std::string& file_path);
void SaveXSet(const std::vector<yacl::crypto::EcPoint>& XSet,
const std::string& file_path,
const std::unique_ptr<yacl::crypto::EcGroup>& ec_group);
Expand Down
4 changes: 3 additions & 1 deletion yacl/examples/sse/sse_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,9 @@ TEST_F(SseTest, SaveAndLoadEDB) {

// 保存EDB到文件
std::string test_dir = "/tmp/sse_test_data/";
system(("mkdir -p " + test_dir).c_str());
if (system(("mkdir -p " + test_dir).c_str()) != 0) {
FAIL() << "Failed to create directory: " << test_dir;
}

auto [k_map, tset, xset] = sse_->SaveEDB(
test_dir + "K_map.bin", test_dir + "TSet.bin", test_dir + "XSet.bin");
Expand Down
2 changes: 1 addition & 1 deletion yacl/examples/sse/tset_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -132,4 +132,4 @@ TEST_F(TSetTest, CompleteRetrievalFlow) {
EXPECT_EQ(retrieved[1].second, "value2");
}

} // namespace yacl::examples::sse
} // namespace examples::sse

0 comments on commit e30437e

Please sign in to comment.