From 4f9360040248d575d923e4688e8c32007a84745a Mon Sep 17 00:00:00 2001
From: Young Joon Lee <entelecheia@hotmail.com>
Date: Fri, 28 Jul 2023 12:05:50 +0900
Subject: [PATCH] feat(hyfi): add GENERATOR to HyFi class

---
 src/hyfi/main/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/hyfi/main/main.py b/src/hyfi/main/main.py
index f5b29d19..1b6b9948 100644
--- a/src/hyfi/main/main.py
+++ b/src/hyfi/main/main.py
@@ -10,7 +10,7 @@
 from omegaconf import DictConfig, OmegaConf
 
 from hyfi.cached_path import cached_path
-from hyfi.composer import Composer, SpecialKeys
+from hyfi.composer import GENERATOR, Composer, SpecialKeys
 from hyfi.copier import Copier
 from hyfi.core import (
     __app_version__,
@@ -84,6 +84,7 @@ class HyFI(
     DATASETs,
     ENVs,
     FUNCs,
+    GENERATOR,
     GPUs,
     GRAPHICs,
     IOLIBs,