ripple-down-rules 0.6.17__py3-none-any.whl → 0.6.19__py3-none-any.whl

ripple_down_rules/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.6.17"
+__version__ = "0.6.19"
 
 import logging
 logger = logging.Logger("rdr")

ripple_down_rules/rdr_decorators.py

@@ -34,7 +34,8 @@ class RDRDecorator:
                  package_name: Optional[str] = None,
                  use_generated_classifier: bool = False,
                  ask_now: Callable[Dict[str, Any], bool] = lambda _: True,
-                 fitting_decorator: Optional[Callable] = None):
+                 fitting_decorator: Optional[Callable] = None,
+                 generate_dot_file: bool = False) -> None:
         """
         :param models_dir: The directory to save/load the RDR models.
         :param output_type: The type of the output. This is used to create the RDR model.
@@ -51,6 +52,10 @@ class RDRDecorator:
         :param viewer: The viewer to use for the RDR model. If None, no viewer will be used.
         :param package_name: The package name to use for relative imports in the RDR model.
         :param use_generated_classifier: If True, the function will use the generated classifier instead of the RDR model.
+        :param ask_now: A callable that takes the case dictionary and returns True if the user should be asked for
+            the output, or False if the function should return the output without asking.
+        :param fitting_decorator: A decorator to use for the fitting function. If None, no decorator will be used.
+        :param generate_dot_file: If True, the RDR model will generate a dot file for visualization.
         :return: A decorator to use a GeneralRDR as a classifier that monitors and modifies the function's output.
         """
         self.rdr_models_dir = models_dir
@@ -69,6 +74,7 @@ class RDRDecorator:
         self.ask_now = ask_now
         self.fitting_decorator = fitting_decorator if fitting_decorator is not None else \
             lambda f: f  # Default to no fitting decorator
+        self.generate_dot_file = generate_dot_file
         self.load()
 
     def decorator(self, func: Callable) -> Callable:
@@ -110,6 +116,8 @@ class RDRDecorator:
                     output = self.generated_classifier(case)
                 else:
                     output = self.rdr.classify(case)
+                    if self.generate_dot_file:
+                        self.rdr.render_evaluated_rule_tree(self.rdr_models_dir + f'/{self.model_name}')
 
             if self.output_name in output:
                 return output[self.output_name]

{ripple_down_rules-0.6.17.dist-info → ripple_down_rules-0.6.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ripple_down_rules
-Version: 0.6.17
+Version: 0.6.19
 Summary: Implements the various versions of Ripple Down Rules (RDR) for knowledge representation and reasoning.
 Author-email: Abdelrhman Bassiouny <abassiou@uni-bremen.de>
 License:                     GNU GENERAL PUBLIC LICENSE

{ripple_down_rules-0.6.17.dist-info → ripple_down_rules-0.6.19.dist-info}/RECORD

@@ -1,8 +1,8 @@
-ripple_down_rules/__init__.py,sha256=hLNcj1JDw69kc8TvVCcxJo2FAvuEYitwgRoV5VS4fis,100
+ripple_down_rules/__init__.py,sha256=SI5OK8zlaHJ4LnHNQ1hp0L91Xem5KQsF_6uS7NrGzVU,100
 ripple_down_rules/experts.py,sha256=4-dMIVeMzFXCLYl_XBG_P7_Xs4sZih9-vZxCIPri6dA,12958
 ripple_down_rules/helpers.py,sha256=RUdfiSWMZjGwCxuCy44TcEJf2UNAFlPJusgHzuAs6qI,4583
 ripple_down_rules/rdr.py,sha256=mtY_kBOjUXRZYz4QFaKN6C2O7liFpbMdnjUnFFnXbdU,57550
-ripple_down_rules/rdr_decorators.py,sha256=j_bIv30nonWz-D5em7dYnV3htlklar2QwFRa5o48XYM,10707
+ripple_down_rules/rdr_decorators.py,sha256=fxoMMAZB1HTfEsuzfIdo3b7PiPB_-AAjD8hy4L0sawQ,11389
 ripple_down_rules/rules.py,sha256=ULIHRbVfKd2L3zzPnssmaeqInWHifPvwwa16MyzakPc,23548
 ripple_down_rules/start-code-server.sh,sha256=otClk7VmDgBOX2TS_cjws6K0UwvgAUJhoA0ugkPCLqQ,949
 ripple_down_rules/utils.py,sha256=uOa_b3FoDnoTLp84RnMbeSyiTMgREtUOkOImQj8B5HQ,64204
@@ -17,8 +17,8 @@ ripple_down_rules/user_interface/ipython_custom_shell.py,sha256=yp-F8YRWGhj1PLB3
 ripple_down_rules/user_interface/object_diagram.py,sha256=FEa2HaYR9QmTE6NsOwBvZ0jqmu3DKyg6mig2VE5ZP4Y,4956
 ripple_down_rules/user_interface/prompt.py,sha256=JceEUGYsd0lIvd-v2y3D3swoo96_C0lxfp3CxM7Vfts,8900
 ripple_down_rules/user_interface/template_file_creator.py,sha256=kwBbFLyN6Yx2NTIHPSwOoytWgbJDYhgrUOVFw_jkDQ4,13522
-ripple_down_rules-0.6.17.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-ripple_down_rules-0.6.17.dist-info/METADATA,sha256=TGxV-tKDfr5ZCcJSStSBxgpcyKEm5yQm6LuBOpDqStI,48294
-ripple_down_rules-0.6.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ripple_down_rules-0.6.17.dist-info/top_level.txt,sha256=VeoLhEhyK46M1OHwoPbCQLI1EifLjChqGzhQ6WEUqeM,18
-ripple_down_rules-0.6.17.dist-info/RECORD,,
+ripple_down_rules-0.6.19.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
+ripple_down_rules-0.6.19.dist-info/METADATA,sha256=rei_pvqow5hMMQOqBDAq-CxM9HTXXrxZ5f-knn-5zms,48294
+ripple_down_rules-0.6.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ripple_down_rules-0.6.19.dist-info/top_level.txt,sha256=VeoLhEhyK46M1OHwoPbCQLI1EifLjChqGzhQ6WEUqeM,18
+ripple_down_rules-0.6.19.dist-info/RECORD,,