| 177 | } |
| 178 | |
| 179 | void SetPairs(const TPathWithScheme& pairsPath, TDatasetSubset loadSubset, IDatasetVisitor* visitor) { |
| 180 | DumpMemUsage("After data read"); |
| 181 | if (pairsPath.Inited()) { |
| 182 | auto pairsDataLoader = GetProcessor<IPairsDataLoader>( |
| 183 | pairsPath, |
| 184 | TPairsDataLoaderArgs{pairsPath, loadSubset} |
| 185 | ); |
| 186 | if (pairsDataLoader->NeedGroupIdToIdxMap()) { |
| 187 | auto maybeGroupIds = visitor->GetGroupIds(); |
| 188 | CB_ENSURE(maybeGroupIds, "Cannot load pairs data with group ids for a dataset without groups"); |
| 189 | pairsDataLoader->SetGroupIdToIdxMap(*maybeGroupIds); |
| 190 | } |
| 191 | pairsDataLoader->Do(visitor); |
| 192 | } |
| 193 | } |
| 194 | |
| 195 | void SetGraph(const TPathWithScheme& graphPath, TDatasetSubset loadSubset, IDatasetVisitor* visitor) { |
| 196 | DumpMemUsage("Before data read graphPath"); |
no test coverage detected