From ae066413890d28631249b0ed35c2c8cbcf22869a Mon Sep 17 00:00:00 2001
From: Surag <10693069+suragnair@users.noreply.github.com>
Date: Fri, 24 Mar 2023 18:41:44 -0700
Subject: [PATCH] Fix to get rc averaging to work for prediction

---
 bpnet/generators/generators.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/bpnet/generators/generators.py b/bpnet/generators/generators.py
index 60514fd..17ce3c1 100755
--- a/bpnet/generators/generators.py
+++ b/bpnet/generators/generators.py
@@ -1208,6 +1208,7 @@ def _generate_batch(self, coords):
             # compute metrics
             inputs['true_profiles'] = profile_predictions
             inputs['true_logcounts'] = logcounts_predictions
+            inputs['rev_comp'] = coords['rev_comp'].values # send this for rev comp averaging
             return inputs