Skip to content

Commit 0ed7d25

Browse files
authored
Merge pull request #19 from alibuild/alibot-cleanup-14069
Please consider the following formatting changes to #14069
2 parents 566ddb7 + a9c33b5 commit 0ed7d25

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.cxx

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,15 +27,16 @@ GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost(const GPUSettingsProcessingNNcl
2727
init(settings);
2828
}
2929

30-
void GPUTPCNNClusterizerHost::loadFromCCDB(std::map<std::string, std::string> settings) {
30+
void GPUTPCNNClusterizerHost::loadFromCCDB(std::map<std::string, std::string> settings)
31+
{
3132
o2::ccdb::CcdbApi ccdbApi;
3233
ccdbApi.init(settings["nnCCDBURL"]);
3334

3435
metadata["inputDType"] = settings["inputDType"];
3536
metadata["outputDType"] = settings["outputDType"];
36-
metadata["nnCCDBEvalType"] = settings["nnCCDBEvalType"]; // classification_1C, classification_2C, regression_1C, regression_2C
37+
metadata["nnCCDBEvalType"] = settings["nnCCDBEvalType"]; // classification_1C, classification_2C, regression_1C, regression_2C
3738
metadata["nnCCDBWithMomentum"] = settings["nnCCDBWithMomentum"]; // 0, 1 -> Only for regression model
38-
metadata["nnCCDBLayerType"] = settings["nnCCDBLayerType"]; // FC, CNN
39+
metadata["nnCCDBLayerType"] = settings["nnCCDBLayerType"]; // FC, CNN
3940
if (settings["nnCCDBInteractionRate"] != "" && std::stoi(settings["nnCCDBInteractionRate"]) > 0) {
4041
metadata["nnCCDBInteractionRate"] = settings["nnCCDBInteractionRate"];
4142
}
@@ -58,16 +59,15 @@ void GPUTPCNNClusterizerHost::init(const GPUSettingsProcessingNNclusterizer& set
5859
std::string class_model_path = settings.nnClassificationPath, reg_model_path = settings.nnRegressionPath;
5960
std::vector<std::string> reg_model_paths;
6061

61-
if(settings.nnLoadFromCCDB) {
62+
if (settings.nnLoadFromCCDB) {
6263
std::map<std::string, std::string> ccdbSettings = {
6364
{"nnCCDBURL", settings.nnCCDBURL},
6465
{"nnCCDBPath", settings.nnCCDBPath},
6566
{"inputDType", settings.nnInferenceInputDType},
6667
{"outputDType", settings.nnInferenceOutputDType},
6768
{"nnCCDBWithMomentum", std::to_string(settings.nnCCDBWithMomentum)},
6869
{"nnCCDBBeamType", settings.nnCCDBBeamType},
69-
{"nnCCDBInteractionRate", std::to_string(settings.nnCCDBInteractionRate)}
70-
};
70+
{"nnCCDBInteractionRate", std::to_string(settings.nnCCDBInteractionRate)}};
7171

7272
std::string nnFetchFolder = "";
7373
std::vector<std::string> fetchMode = o2::utils::Str::tokenize(settings.nnCCDBFetchMode, ':');

GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,9 @@ class GPUTPCNNClusterizerHost
4949
o2::ml::OrtModel model_class, model_reg_1, model_reg_2; // For splitting clusters
5050
std::vector<std::string> reg_model_paths;
5151

52-
private:
53-
std::map<std::string, std::string> metadata;
54-
std::map<std::string, std::string> headers;
52+
private:
53+
std::map<std::string, std::string> metadata;
54+
std::map<std::string, std::string> headers;
5555
}; // class GPUTPCNNClusterizerHost
5656

5757
} // namespace o2::gpu

0 commit comments

Comments
 (0)