FEAT-modin-project#2363: fix index name setter in OmniSci backend

Signed-off-by: ienkovich <ilya.enkovich@intel.com>
gshimansky · Nov 10, 2020 · 18249a7 · 18249a7
1 parent 12debef
commit 18249a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/modin/experimental/engines/omnisci_on_ray/frame/data.py b/modin/experimental/engines/omnisci_on_ray/frame/data.py
@@ -1251,11 +1251,11 @@ def set_index_name(self, name):
             return self
 
         names = self._mangle_index_names([name])
+        exprs = OrderedDict()
         if self._index_cols is None:
-            exprs = OrderedDict()
-            exprs[name] = self.ref("__rowid__")
+            exprs[names[0]] = self.ref("__rowid__")
         else:
-            exprs = self._index_exprs()
+            exprs[names[0]] = self.ref(self._index_cols[0])
 
         for col in self.columns:
             exprs[col] = self.ref(col)