From c67bd9ebefde9dda0aaf4a12345d179d6ae69c95 Mon Sep 17 00:00:00 2001
From: Harsha <hvgazula@umich.edu>
Date: Tue, 12 Mar 2024 13:49:54 -0400
Subject: [PATCH] resolved https://github.com/neuronets/nobrainer/issues/278

---
 nobrainer/dataset.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nobrainer/dataset.py b/nobrainer/dataset.py
index 791737fd..30c5be53 100644
--- a/nobrainer/dataset.py
+++ b/nobrainer/dataset.py
@@ -121,6 +121,9 @@ def from_tfrecords(
         )
         block_length = len([0 for _ in first_shard])
 
+        if not n_volumes:
+            n_volumes = block_length * len(files)
+
         dataset = dataset.interleave(
             map_func=lambda x: tf.data.TFRecordDataset(
                 x, compression_type=compression_type