Add p argument to RandAugment and TrivialAugmentWide and fix bug in RandomApply

[james77777778]

No description provided.

[codecov]

Codecov Report

Attention: Patch coverage is 71.42857% with 8 lines in your changes missing coverage. Please review.

Project coverage is 84.45%. Comparing base (e3dedff) to head (681c0c3).

Files	Patch %	Lines
keras_aug/_src/layers/vision/rand_augment.py	66.66%	4 Missing ⚠️
keras_aug/_src/layers/vision/trivial_augment.py	66.66%	4 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #134      +/-   ##
==========================================
- Coverage   84.50%   84.45%   -0.06%     
==========================================
  Files          44       44              
  Lines        3795     3814      +19     
==========================================
+ Hits         3207     3221      +14     
- Misses        588      593       +5     

Flag	Coverage Δ
keras-aug	84.45% <71.42%> (-0.06%)	⬇️
keras-aug-jax	84.13% <71.42%> (-0.06%)	⬇️
keras-aug-tensorflow	83.29% <71.42%> (-0.05%)	⬇️
keras-aug-torch	84.32% <71.42%> (-0.06%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

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