pgroup debugging

This commit is contained in:
David
2023-12-10 23:00:46 -08:00
parent 7b163e0d08
commit a58b45b415
+13
View File
@@ -3,6 +3,8 @@ import torch
import numpy as np import numpy as np
from PIL.Image import Image from PIL.Image import Image
import keyframed as kf
logging.basicConfig(level=logging.INFO, logging.basicConfig(level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s') format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -36,6 +38,10 @@ def _inspect(item, depth=0):
if isinstance(item, Image): if isinstance(item, Image):
logger.info(f"{pad}item.mode: {item.mode}") logger.info(f"{pad}item.mode: {item.mode}")
if isinstance(item, kf.ParameterGroup):
logger.info(f"{pad}item.parameters: {item.parameters}")
# to do: be fancy and change to a match statement # to do: be fancy and change to a match statement
@@ -147,6 +153,13 @@ class KfDebug_Curve(KfDebug_Passthrough):
RETURN_TYPES = ("KEYFRAMED_CURVE",) RETURN_TYPES = ("KEYFRAMED_CURVE",)
class KfDebug_PGroup(KfDebug_Passthrough):
RETURN_TYPES = ("PARAMETER_GROUP",)
class KfDebug_Schedule(KfDebug_Passthrough):
RETURN_TYPES = ("SCHEDULE",)
# ########################### # ###########################