[Frontend] Add support for logical or, and and not in autograph

[sergei-mironov]

Support Python and-, or- and not- operators in Autograph-converted programs.

[sc-41303]

Test cases mentioned:

• cases where the boolean statement is a mix of dynamic and static variables,
• test cases where the parameters are of different types (jax arrays, booleans, numpy arrays)
• cases where the boolean is used as conditionals for if statements
• cases where the boolean is used as the conditional for while statemen
• test cases where the parameters are arbitrary Python objects which can decay to Boolean (like strings, lists, etc, anything that Python allows in an if statement)

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (bc8ad36) 99.62% compared to head (50f56fe) 99.62%.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #325   +/-   ##
=======================================
  Coverage   99.62%   99.62%           
=======================================
  Files          42       42           
  Lines        7438     7454   +16     
  Branches      434      439    +5     
=======================================
+ Hits         7410     7426   +16     
  Misses         14       14           
  Partials       14       14           

Files	Coverage Δ
frontend/catalyst/ag_primitives.py	100.00% <100.00%> (ø)
frontend/catalyst/autograph.py	100.00% <100.00%> (ø)

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[sergei-mironov]

Hi @josh146. I expect this logic ops implementation to have corner-cases, so I 'd like to know what do we want to support in the first place here. Do we have any particular use-cases in mind, e.g. among the tutorial algorithms which I could check?

[dime10]

Looks good 👍