From c63ec817501bf188abd38a4fc3797af8b829aa20 Mon Sep 17 00:00:00 2001
From: Darijan Gudelj <darijan@fb.com>
Date: Tue, 2 Aug 2022 09:09:47 -0700
Subject: [PATCH] fix eval_batches in V1

Summary: fix to the D38275943 (https://github.com/facebookresearch/pytorch3d/commit/597e0259dc43bf4903e9c99f5d61410c1ad75b78).

Reviewed By: bottler

Differential Revision: D38355683

fbshipit-source-id: f326f45279fafa57f24b9211ebd3fda18a518937
---
 pytorch3d/implicitron/dataset/json_index_dataset.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pytorch3d/implicitron/dataset/json_index_dataset.py b/pytorch3d/implicitron/dataset/json_index_dataset.py
index 115514da..ae08ae36 100644
--- a/pytorch3d/implicitron/dataset/json_index_dataset.py
+++ b/pytorch3d/implicitron/dataset/json_index_dataset.py
@@ -177,7 +177,9 @@ class JsonIndexDataset(DatasetBase, ReplaceableBase):
                 raise ValueError(
                     "Cannot define both eval_batch_index and eval_batches."
                 )
-            self.eval_batches = self.seq_frame_index_to_dataset_index()
+            self.eval_batches = self.seq_frame_index_to_dataset_index(
+                self.eval_batch_index
+            )
 
     def is_filtered(self):
         """