update (#268)

Author: rusty1s

csrc/cpu/neighbor_sample_cpu.cpp

@@ -365,11 +365,11 @@ hetero_sample(const vector<node_t> &node_types,
 
     if (temporal) {
       for (const auto &kv : temp_samples_dict) {
-        slice_dict[kv.first] = {0, kv.second.size()};
+        slice_dict[kv.first] = {slice_dict.at(kv.first).second, kv.second.size()};
       }
     } else {
       for (const auto &kv : samples_dict)
-        slice_dict[kv.first] = {0, kv.second.size()};
+        slice_dict[kv.first] = {slice_dict.at(kv.first).second, kv.second.size()};
     }
   }
 
@@ -504,4 +504,4 @@ hetero_temporal_neighbor_sample_cpu(
         node_types, edge_types, colptr_dict, row_dict, input_node_dict,
         num_neighbors_dict, node_time_dict, num_hops);
   }
-}
+}