add import lightning if needed

blei-lab · Oct 16, 2024 · 2068b76 · 2068b76
1 parent 13f10f5
commit 2068b76
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/testbed/src/testbed/__main__.py b/testbed/src/testbed/__main__.py
@@ -1,6 +1,8 @@
 # noinspection PyUnresolvedReferences
 import lightgbm as lgb  # noqa F401
 
+# import lightning
+
 import argparse
 import logging
 import os