fix(csvts/torch): allow to read csv timeserie directly from query

jolibrain · Oct 13, 2020 · 76023db · 76023db
1 parent d6749d0
commit 76023db
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/backends/torch/torchinputconns.cc b/src/backends/torch/torchinputconns.cc
@@ -1049,8 +1049,9 @@ namespace dd
             std::vector<at::Tensor> label_sequence;
 
             tstart = seq.size() - _timesteps;
-            _ids.push_back(_fnames[vecindex] + " #" + std::to_string(tstart)
-                           + "_" + std::to_string(tstart + _timesteps - 1));
+            if (_fnames.size() > static_cast<unsigned int>(vecindex))
+              _ids.push_back(_fnames[vecindex] + " #" + std::to_string(tstart)
+                             + "_" + std::to_string(tstart + _timesteps - 1));
             for (int ti = tstart; ti < tstart + _timesteps; ++ti)
               {
                 std::vector<double> datavec;