fix sub-POMDP construction

randriu · Feb 19, 2024 · 8aab0e1 · 8aab0e1
1 parent bec3a74
commit 8aab0e1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/payntbind/src/synthesis/translation/SubPomdpBuilder.cpp b/payntbind/src/synthesis/translation/SubPomdpBuilder.cpp
@@ -76,6 +76,7 @@ namespace synthesis {
             }
 
             auto state = state_translator.retrieve(translated_state);
+            builder.newRowGroup(builder.getLastRow()+1);
             synthesis::translateTransitionMatrixRowGroup(
                 pomdp, builder, state_translator.itemToTranslation(), choice_translator.itemToTranslation(), state
             );

diff --git a/payntbind/src/synthesis/translation/componentTranslations.cpp b/payntbind/src/synthesis/translation/componentTranslations.cpp
@@ -118,7 +118,6 @@ namespace synthesis {
         std::vector<uint64_t> const& original_to_translated_choice,
         uint64_t state
     ) {
-        builder.newRowGroup(builder.getLastRow());
         for(auto const& choice: model.getTransitionMatrix().getRowGroupIndices(state)) {
             synthesis::translateTransitionMatrixRow(
                 model, builder, original_to_translated_state, original_to_translated_choice, choice