update prepare_data_per_node as attributes

Author: ninginthecloud

pytorch_lightning/core/hooks.py

@@ -372,11 +372,9 @@ def configure_sharded_model(self) -> None:
 class DataHooks:
     """Hooks to be used for data related stuff."""
 
-    __jit_unused_properties__ = ["prepare_data_per_node"]
-
     def __init__(self) -> None:
         """
-        property:
+        Attributes:
             prepare_data_per_node:
                 If True, each LOCAL_RANK=0 will call prepare data.
                 Otherwise only NODE_RANK=0, LOCAL_RANK=0 will prepare data.

pytorch_lightning/core/lightning.py

@@ -79,7 +79,6 @@ class LightningModule(
             "truncated_bptt_steps",
             "loaded_optimizer_states_dict",
         ]
-        + DataHooks.__jit_unused_properties__
         + DeviceDtypeModuleMixin.__jit_unused_properties__
         + HyperparametersMixin.__jit_unused_properties__
     )