addressed comments

Author: KshitizLohia

ads/common/decorator/runtime_dependency.py

@@ -64,10 +64,9 @@ class OptionalDependency:
     OPTUNA = "oracle-ads[optuna]"
     SPARK = "oracle-ads[spark]"
     HUGGINGFACE = "oracle-ads[huggingface]"
-    GREAT_EXPECTATIONS = "oracle-ads[great-expectations]"
+    FEATURE_STORE = "oracle-ads[feature-store]"
     GRAPHVIZ = "oracle-ads[graphviz]"
     MLM_INSIGHTS = "oracle-ads[mlm_insights]"
-    PYARROW = "oracle-ads[pyarrow]"
 
 
 def runtime_dependency(

ads/feature_store/data_validation/great_expectation.py

@@ -21,8 +21,8 @@
     from great_expectations.validator.validator import Validator
 except ModuleNotFoundError:
     raise ModuleNotFoundError(
-        f"The `great-expectations` module was not found. Please run `pip install "
-        f"{OptionalDependency.GREAT_EXPECTATIONS}`."
+        f"The `feature-store` module was not found. Please run `pip install "
+        f"{OptionalDependency.FEATURE_STORE}`."
     )
 except Exception as e:
     raise

setup.py

@@ -72,9 +72,13 @@
     "bds": ["ibis-framework[impala]", "hdfs[kerberos]", "sqlalchemy"],
     "spark": ["pyspark>=3.0.0", "delta-spark"],
     "huggingface": ["transformers"],
-    "great-expectations": ["great-expectations==0.15.39"],
+    "feature-store": [
+        "pyspark>=3.0.0",
+         "delta-spark",
+         "great-expectations==0.15.39",
+         "pyarrow"
+    ],
     "mlm_insights": ["mlm_insights==0.1.0.dev1"],
-    "pyarrow": ["pyarrow"],
 }
 
 this_directory = Path(__file__).parent