🔨 Fix Arbitrary imports in get_datamodule() (#2736)

* fix semgrep in get_datamodule()

Signed-off-by: Rajesh Gangireddy <rajesh.gangireddy@intel.com>

* Update CHANGELOG.md

Signed-off-by: Rajesh Gangireddy <rajesh.gangireddy@intel.com>

---------

Signed-off-by: Rajesh Gangireddy <rajesh.gangireddy@intel.com>

Author: rajeshgangireddy

CHANGELOG.md

@@ -16,7 +16,8 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
 ### Changed
 
-- Update import location for `get_exportable_transform` from `anomalib.utils.transform` to `anomalib.pre_processing.utils.transform`. This update aligns with the reorganization of the `anomalib` project in `v2.0.0`, and the previous path no longer exists.
+- 🔨Fix semgrep security issue for `get_datamodule()` method in https://github.com/open-edge-platform/anomalib/pull/2736
+- 🔨Update doc for `get_exportable_transform()` method in https://github.com/open-edge-platform/anomalib/pull/2731
 
 ### Deprecated
 

src/anomalib/data/__init__.py

@@ -119,25 +119,29 @@ def get_datamodule(config: DictConfig | ListConfig | dict) -> AnomalibDataModule
         ... })
         >>> datamodule = get_datamodule(config)
     """
-    logger.info("Loading the datamodule")
+    logger.info("Loading the datamodule and dataset class from the config.")
 
+    # Getting the datamodule class from the config.
     if isinstance(config, dict):
         config = DictConfig(config)
+    _config = config.data if "data" in config else config
+
+    # All the sub data modules are imported to anomalib.data. So need to import the module dynamically using paths.
+    module = importlib.import_module("anomalib.data")
+    data_class_name = _config.class_path.split(".")[-1]
+    # check if the data_class exists in the module
+    if not hasattr(module, data_class_name):
+        logger.error(
+            f"Dataclass '{data_class_name}' not found in module '{module.__name__}'. "
+            f"Available classes are {AnomalibDataModule.__subclasses__()}",
+        )
+        error_str = f"Dataclass '{data_class_name}' not found in module '{module.__name__}'."
+        raise UnknownDatamoduleError(error_str)
+    dataclass = getattr(module, data_class_name)
 
-    try:
-        _config = config.data if "data" in config else config
-        if len(_config.class_path.split(".")) > 1:
-            module = importlib.import_module(".".join(_config.class_path.split(".")[:-1]))
-        else:
-            module = importlib.import_module("anomalib.data")
-    except ModuleNotFoundError as exception:
-        logger.exception(f"ModuleNotFoundError: {_config.class_path}")
-        raise UnknownDatamoduleError from exception
-    dataclass = getattr(module, _config.class_path.split(".")[-1])
     init_args = {**_config.get("init_args", {})}  # get dict
     if "image_size" in init_args:
         init_args["image_size"] = to_tuple(init_args["image_size"])
-
     return dataclass(**init_args)