added flatbuffers to setup.py

neil-tan · neil-tan · commit 5c1bf6050a45 · 2019-12-05T17:02:40.000-08:00
diff --git a/setup.py b/setup.py
@@ -37,7 +37,8 @@
         'torch',
         'torchvision',
         'onnx-tf==1.2.1',
-        'graphviz'
+        'graphviz',
+        'flatbuffers'
     ],
     extras_require={
         'dev': ['pytest']
