From f6e1c45d57a45cf478645ca23d1d2f33128d1edf Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Thu, 21 Apr 2016 19:05:01 +0200
Subject: [PATCH] gen/genlib/record: fix connect

---
 litex/gen/genlib/record.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/litex/gen/genlib/record.py b/litex/gen/genlib/record.py
index e98922ffd..2692ab4b6 100644
--- 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