|
25 | 25 | #include "SimulationDataFormat/BaseHits.h" |
26 | 26 | #include "SimulationDataFormat/StackParam.h" |
27 | 27 | #include "CommonUtils/ConfigurationMacroHelper.h" |
28 | | -#include "CCDB/CcdbApi.h" |
| 28 | +#include "CCDB/BasicCCDBManager.h" |
29 | 29 | #include "ML/OrtInterface.h" |
30 | 30 |
|
31 | 31 | #include "TLorentzVector.h" // for TLorentzVector |
@@ -70,11 +70,15 @@ class OnnxPrimaryTransport |
70 | 70 | throw std::runtime_error("Stack.transportPrimaryOnnxCCDBPath must be configured"); |
71 | 71 | } |
72 | 72 |
|
73 | | - o2::ccdb::CcdbApi ccdb; |
74 | | - ccdb.init(param.transportPrimaryOnnxCCDBUrl); |
| 73 | + auto& ccdbManager = o2::ccdb::BasicCCDBManager::instance(); |
| 74 | + auto& ccdb = ccdbManager.getCCDBAccessor(); |
75 | 75 | std::map<std::string, std::string> headers; |
| 76 | + const auto createdNotAfter = ccdbManager.getCreatedNotAfter(); |
| 77 | + const auto createdNotBefore = ccdbManager.getCreatedNotBefore(); |
76 | 78 | ccdb.loadFileToMemory(mModelBytes, param.transportPrimaryOnnxCCDBPath, {}, |
77 | | - param.transportPrimaryOnnxTimestamp, &headers, {}, {}, {}); |
| 79 | + ccdbManager.getTimestamp(), &headers, {}, |
| 80 | + createdNotAfter ? std::to_string(createdNotAfter) : "", |
| 81 | + createdNotBefore ? std::to_string(createdNotBefore) : ""); |
78 | 82 | if (mModelBytes.empty()) { |
79 | 83 | throw std::runtime_error("failed to retrieve ONNX model from CCDB path " + param.transportPrimaryOnnxCCDBPath); |
80 | 84 | } |
|
0 commit comments