From 89e69f86fe5f317c4b162318f6b99c3fc4a6d29b Mon Sep 17 00:00:00 2001
From: wallkop <120839632@qq.com>
Date: Sat, 29 Jun 2024 19:39:53 +0800
Subject: [PATCH] Fix: key error caused by csv generated by openFace deployed
 on Linux.

---
 nerf_triplane/provider.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nerf_triplane/provider.py b/nerf_triplane/provider.py
index c0e1a77..b55ee4b 100644
--- a/nerf_triplane/provider.py
+++ b/nerf_triplane/provider.py
@@ -251,7 +251,7 @@ def __init__(self, opt, device, type='train', downscale=1):
         if self.opt.au45:
             import pandas as pd
             au_blink_info = pd.read_csv(os.path.join(self.root_path, 'au.csv'))
-            bs = au_blink_info[' AU45_r'].values
+            bs = au_blink_info.get('AU45_r', au_blink_info.get(' AU45_r')).values
         else:
             bs = np.load(os.path.join(self.root_path, 'bs.npy'))
             if self.opt.bs_area == "upper":