gen/genlib/record: fix connect

diff --git a/litex/gen/genlib/record.py b/litex/gen/genlib/record.py
index e98922ffdaa54e6e9a98aaea37a573fdf2469f3c..2692ab4b6124efcadadcdcb122b104b076c17920 100644 (file)
--- a/litex/gen/genlib/record.py
+++ b/litex/gen/genlib/record.py
@@ -136,10 +136,14 @@ class Record:
             _keep = set([f[0] for f in self.layout])
         elif isinstance(keep, list):
             _keep = set(keep)
+        else:
+            _keep = keep
         if omit is None:
             _omit = set()
         elif isinstance(omit, list):
             _omit = set(omit)
+        else:
+            _omit = omit
 
         _keep = _keep - _omit