From e53b090a5cd0361e3c9be6b5b9ed2a61a2238184 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 15 Jun 2020 14:08:15 +0200
Subject: [PATCH] Added prefix in distanceModule values to avoid them being
 treated as generic numbers bu the Dict

---
 torch_modules/src/DistanceModule.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_modules/src/DistanceModule.cpp b/torch_modules/src/DistanceModule.cpp
index f529537..1b8034b 100644
--- a/torch_modules/src/DistanceModule.cpp
+++ b/torch_modules/src/DistanceModule.cpp
@@ -100,7 +100,7 @@ void DistanceModuleImpl::addToContext(std::vector<std::vector<long>> & context,
         long dist = std::abs(config.getRelativeDistance(from, to));
 
         if (dist <= threshold)
-          contextElement.emplace_back(dict.getIndexOrInsert(fmt::format("{}", dist)));
+          contextElement.emplace_back(dict.getIndexOrInsert(fmt::format("distance({})", dist)));
         else
           contextElement.emplace_back(dict.getIndexOrInsert(Dict::unknownValueStr));
       }
-- 
GitLab