Skip to content
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

Implement async generator #648

Open
wants to merge 8 commits into
base: main
Choose a base branch
from

Conversation

JasonKChow
Copy link
Contributor

@JasonKChow JasonKChow commented Feb 10, 2025

Summary: A generator that allows asynchronous generation of points. Uses a different process to generate points. This doesn't fix the situation where modeling fitting takes a long time.

Test Plan: New test

Stack created with Sapling. Best reviewed with ReviewStack.

Summary: Rewrite the server to use asyncio and be multithreaded. Allows handling multiple clients.

Test Plan: Tests updated. Later test also tests multiple db writes simultaneously.
Summary: Use scoped sessions that is shared between threads to allow async db operations.

Test Plan: Tests should pass
Summary: Logging levels were used inconsistently and would cause problems with multithreaded servers. We remove this ability.

Test Plan: Tests updated.
Summary: All message handlers will emit a info logging message to support better live tracking of server functions.

Test Plan: Tests should pass
Summary: To support multi client server, strategy methods will not act on copies of models to avoid changing tensor gradients between two threads.

Test Plan: New test
Summary: A subclass of the aepsych server with methods specifically to run the server in a background process. This will be used to ensured that even within the same main script, the server will run like an actual server and does not do anything sneaky like bypassing the async queue.

Test Plan: New test
Summary: Add the ___reduce__ method to transform wrappers (model/generator). This allows transform wrapped objects to be pickled (with original pickle module, not just dill)

Test Plan: tests should pass
Summary: A generator that allows asynchronous generation of points. Uses a different process to generate points. This doesn't fix the situation where modeling fitting takes a long time.

Test Plan: New test
@facebook-github-bot
Copy link
Contributor

@JasonKChow has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants