Fix flake in keras subclass test (workaround)
@@ -12,7 +12,15 @@ depend:
assert:
- :wandb:runs_len: 1
- :wandb:artifacts[model-lovely-dawn-32][type]: model
- - :wandb:artifacts[model-lovely-dawn-32][num]: 2
+ # Test doesnt always save 2 models since it might not improve
+ # on second epoch. When test is made more determinstic, this
+ # should be changed back to a stricter check
+ - :op:>=:
+ - :wandb:artifacts[model-lovely-dawn-32][num]
+ - 1
+ - :op:<=:
+ - :wandb:artifacts[model-lovely-dawn-32][num]
+ - 2
- :op:contains:
- :wandb:runs[0][telemetry][3] # feature
- 8 # keras
