-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: chex
-ify testsuite
#12
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Very close to merging. Have left some comments.
Let's also add a setUp
function so that we don't need to setup the inputs? What do you think?
jflux/cli.py
Outdated
img = Image.fromarray((127.5 * (x + 1.0))) | ||
img.save(fn, quality=95, subsampling=0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for doing this.
I am doing this
x = (127.5 * (x + 1.0))
x_numpy = np.array(x.astype(jnp.uint8))
img = Image.fromarray(x_numpy)
img.save(fn, quality=95, subsampling=0)
idx += 1
Can you add this bit in this PR?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added in 07cf60a
jflux/util.py
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
in port_flux
we are missing a return statement
Could you add return flux
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added in 9bd82d6
@@ -32,7 +32,10 @@ jflux = { workspace = true } | |||
flux = { git = "https://github.com/black-forest-labs/flux.git" } | |||
|
|||
[tool.ruff.lint] | |||
select = ["I001"] | |||
select = ["E", "F", "I001", "W"] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just asking -- what does this do?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So previously, we weren't utilising most of the nice ruff
rules. This snippet enables some nice rules from https://docs.astral.sh/ruff/rules/
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🥳
.pre-commit-config.yaml
.README
instructions for running the cli..vscode/settings.json
with a simple vscode config.jflux/__main__.py
.E
,F
andW
.chex
-ify testsuite.