Skip to content

Conversation

@karenfeng
Copy link
Collaborator

@karenfeng karenfeng commented Oct 15, 2019

What changes are proposed in this pull request?

  • Make sure tests pass with Java (not Kryo) serialization
  • Move README from md to rst to play nice with pip descriptions
  • Prune (mostly GATK) dependencies to reduce package resolution time
  • Set up pipeline to release maven artifacts to Bintray
  • Set up Python wheel creation (note that typeguard exists in PyPi but not TestPyPi)

Instructions

  • To create a maven artifact and...
  • To create a wheel and...
    • Publish to TestPyPi
      • Prereq: update your conda environment so that you can use setuptools and twine
      • Run python setup.py bdist_wheel from glow/python
      • Run twine upload --repository-url https://test.pypi.org/legacy/ dist/*
    • Promote to PyPi
      • Run twine upload dist/*

How is this patch tested?

  • Unit tests
  • Integration tests
  • Manual tests

A DBR cluster was set up with the maven artifacts and wheel, and basic functionality was tested.

@karenfeng karenfeng force-pushed the pypy-attempt2 branch 2 times, most recently from 96aeb4c to 04fb64a Compare October 15, 2019 15:04
Copy link
Contributor

@henrydavidge henrydavidge left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! I had a bunch of kind of nit-picky comments. Btw, we'll need to document at least internally the release process.

version.sbt Outdated
@@ -0,0 +1 @@
version in ThisBuild := "0.1.1-SNAPSHOT"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why have the version in a separate file? Also the version should be 0.1.0, as it is in build.sbt.

Copy link
Collaborator Author

@karenfeng karenfeng Oct 16, 2019

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is automatically generated as part of the sbt release process: https://github.com/sbt/sbt-release

@karenfeng karenfeng force-pushed the pypy-attempt2 branch 2 times, most recently from 591dfa5 to c42780a Compare October 16, 2019 08:00
Copy link
Contributor

@henrydavidge henrydavidge left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
Signed-off-by: Karen Feng <[email protected]>
@karenfeng karenfeng merged commit ad765e6 into projectglow:master Oct 16, 2019
henrydavidge pushed a commit to henrydavidge/glow that referenced this pull request Jun 22, 2020
* Added necessary modifications to accomodate covariates in model fitting.

The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X.  This PR makes numerous changes to accomodate covariate matrix C.

Adding covariates required the following breaking changes to the APIs:
 * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform():
   * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf)
   * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf)

Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument.

Two new tests have been added to test_ridge_regression.py to test run modes with covariates:
 * test_ridge_reducer_transform_with_cov
 * test_two_level_regression_with_cov

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Cleaned up one unnecessary Pandas import
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Small changes for clarity and consistence with the rest of the code.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Forgot one usage of coalesce
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Added a couple of comments to explain logic and replaced usages of .values with .array
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Fixed one instance of the change .values -> .array where it was made in error.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Typo in test_ridge_regression.py.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Style auto-updates with yapfAll
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

Co-authored-by: Leland Barnard <[email protected]>
Co-authored-by: Karen Feng <[email protected]>
henrydavidge pushed a commit to henrydavidge/glow that referenced this pull request Jun 22, 2020
* Run scalastyle as part of tests

* Update build.sbt

Signed-off-by: Henry Davidge <[email protected]>
henrydavidge pushed a commit to henrydavidge/glow that referenced this pull request Jun 22, 2020
* Reapply patch

Signed-off-by: Karen Feng <[email protected]>

* Update setup.py

Signed-off-by: Karen Feng <[email protected]>

* Path

Signed-off-by: Karen Feng <[email protected]>

* Setting version to 0.1.0

Signed-off-by: Karen Feng <[email protected]>

* Pypi version

Signed-off-by: Karen Feng <[email protected]>

* Setting version to 0.1.1-SNAPSHOT

Signed-off-by: Karen Feng <[email protected]>

* Update README

Signed-off-by: Karen Feng <[email protected]>

* Address comments

Signed-off-by: Karen Feng <[email protected]>

* Rename python lib

Signed-off-by: Karen Feng <[email protected]>

* Setting version to 0.1.0

Signed-off-by: Karen Feng <[email protected]>

* Modify domain

Signed-off-by: Karen Feng <[email protected]>

* One more domain update

Signed-off-by: Karen Feng <[email protected]>

* Remove version from build.sbt

Signed-off-by: Karen Feng <[email protected]>

* Change domain

Signed-off-by: Karen Feng <[email protected]>

Signed-off-by: Henry Davidge <[email protected]>
henrydavidge pushed a commit to henrydavidge/glow that referenced this pull request Jun 22, 2020
* Added necessary modifications to accomodate covariates in model fitting.

The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X.  This PR makes numerous changes to accomodate covariate matrix C.

Adding covariates required the following breaking changes to the APIs:
 * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform():
   * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf)
   * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf)

Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument.

Two new tests have been added to test_ridge_regression.py to test run modes with covariates:
 * test_ridge_reducer_transform_with_cov
 * test_two_level_regression_with_cov

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Cleaned up one unnecessary Pandas import
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Small changes for clarity and consistence with the rest of the code.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Forgot one usage of coalesce
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Added a couple of comments to explain logic and replaced usages of .values with .array
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Fixed one instance of the change .values -> .array where it was made in error.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Typo in test_ridge_regression.py.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Style auto-updates with yapfAll
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

Co-authored-by: Leland Barnard <[email protected]>
Co-authored-by: Karen Feng <[email protected]>
Signed-off-by: Henry Davidge <[email protected]>
henrydavidge added a commit that referenced this pull request Jun 22, 2020
* Add Leland's demo notebook

* block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2)

* blocks

Signed-off-by: kianfar77 <[email protected]>

* test vcf

Signed-off-by: kianfar77 <[email protected]>

* transformer

Signed-off-by: kianfar77 <[email protected]>

* remove extra

Signed-off-by: kianfar77 <[email protected]>

* refactor and conform with ridge namings

Signed-off-by: kianfar77 <[email protected]>

* test

Signed-off-by: kianfar77 <[email protected]>

* test files

Signed-off-by: kianfar77 <[email protected]>

* remove extra file

Signed-off-by: kianfar77 <[email protected]>

* sort_key

Signed-off-by: kianfar77 <[email protected]>

* feat: ridge models for wgr added (#1)

* feat: ridge models for wgr added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Doc strings added for levels/functions.py
Some typos fixed in ridge_model.py
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* ridge_model and RidgeReducer unit tests added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* RidgeRegression unit tests added
test data README added
ridge_udfs.py docstrings added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Changes made to accessing the sample ID map and more docstrings

The map_normal_eqn and score_models functions previously expected the
sample IDs for a given sample block to be found in the Pandas DataFrame,
which mean we had to join them on before the .groupBy().apply().  These
functions now expect the sample block to sample IDs mapping to be
provided separately as a dict, so that the join is no longer required.
RidgeReducer and RidgeRegression APIs remain unchanged.

docstrings have been added for RidgeReducer and RidgeRegression classes.

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Refactored object names and comments to reflect new terminology

Where 'block' was previously used to refer to the set of columns in a
block, we now use 'header_block'
Where 'group' was previously used to refer to the set of samples in a
block, we now use 'sample_block'

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* address comments

Signed-off-by: Karen Feng <[email protected]>

* Test fixup

Signed-off-by: Karen Feng <[email protected]>

* Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching

Signed-off-by: Karen Feng <[email protected]>

* PyArrow 0.15.1 only with PySpark 3

Signed-off-by: Karen Feng <[email protected]>

* Don't use toPandas()

Signed-off-by: Karen Feng <[email protected]>

* Upgrade pyarrow

Signed-off-by: Karen Feng <[email protected]>

* Only register once

Signed-off-by: Karen Feng <[email protected]>

* Minimize memory usage

Signed-off-by: Karen Feng <[email protected]>

* Select before head

Signed-off-by: Karen Feng <[email protected]>

* set up/tear down

Signed-off-by: Karen Feng <[email protected]>

* Try limiting pyspark memory

Signed-off-by: Karen Feng <[email protected]>

* No teardown

Signed-off-by: Karen Feng <[email protected]>

* Extend timeout

Signed-off-by: Karen Feng <[email protected]>

* Simplify ordering logic in levels code (#7)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* start changing for readability

* use input label ordering

* rename create_row_indexer

* undo column sort

* change reduce

Signed-off-by: Henry D <[email protected]>

* further simplify reduce

* sorted alpha names

* remove ordering

* comments

Signed-off-by: Henry D <[email protected]>

* Set arrow env var in build

Signed-off-by: Henry D <[email protected]>

* faster sort

* add test file

* undo test data change

* >=

* formatting

* empty

Co-authored-by: Karen Feng <[email protected]>

* Limit Spark memory conf in tests (#9)

* yapf

Signed-off-by: Karen Feng <[email protected]>

* yapf transform

Signed-off-by: Karen Feng <[email protected]>

* Set driver memory

Signed-off-by: Karen Feng <[email protected]>

* Try changing spark mem

Signed-off-by: Karen Feng <[email protected]>

* match java tests

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* remove driver memory flag

Signed-off-by: Karen Feng <[email protected]>

* Improve partitioning in block_variants_and_samples transformer (#11)

Signed-off-by: kianfar77 <[email protected]>

* Remove unnecessary header_block grouping (#10)

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* Create sample ID blocking helper functions (#12)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* simplify tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* yapf

Signed-off-by: Karen Feng <[email protected]>

* index map compat

Signed-off-by: Karen Feng <[email protected]>

* Add docs

Signed-off-by: Karen Feng <[email protected]>

* Add more tests

Signed-off-by: Karen Feng <[email protected]>

* pass args as ints

Signed-off-by: Karen Feng <[email protected]>

* Don't roll our own splitter

Signed-off-by: Karen Feng <[email protected]>

* rename sample_index to sample_blocks

Signed-off-by: Karen Feng <[email protected]>

* Add type-checking to WGR APIs (#14)

* Add type-checking to APIs

Signed-off-by: Karen Feng <[email protected]>

* Check valid alphas

Signed-off-by: Karen Feng <[email protected]>

* check 0 sig

Signed-off-by: Karen Feng <[email protected]>

* Add to install_requires list

Signed-off-by: Karen Feng <[email protected]>

* cleanup comments

Signed-off-by: Karen Feng <[email protected]>

* Add covariate support (#13)

* Added necessary modifications to accomodate covariates in model fitting.

The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X.  This PR makes numerous changes to accomodate covariate matrix C.

Adding covariates required the following breaking changes to the APIs:
 * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform():
   * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf)
   * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf)

Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument.

Two new tests have been added to test_ridge_regression.py to test run modes with covariates:
 * test_ridge_reducer_transform_with_cov
 * test_two_level_regression_with_cov

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Cleaned up one unnecessary Pandas import
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Small changes for clarity and consistence with the rest of the code.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Forgot one usage of coalesce
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Added a couple of comments to explain logic and replaced usages of .values with .array
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Fixed one instance of the change .values -> .array where it was made in error.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Typo in test_ridge_regression.py.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Style auto-updates with yapfAll
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

Co-authored-by: Leland Barnard <[email protected]>
Co-authored-by: Karen Feng <[email protected]>

* Flatten estimated phenotypes (#15)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Clean up tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Order to match labeldf

Signed-off-by: Karen Feng <[email protected]>

* Check we tie-break

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* test var name

Signed-off-by: Karen Feng <[email protected]>

* clean up tests

Signed-off-by: Karen Feng <[email protected]>

* Clean up docs

Signed-off-by: Karen Feng <[email protected]>

* Add fit_transform function to models (#17)

Signed-off-by: Karen Feng <[email protected]>

* Rename levels (#20)

* Rename levels to wgr

Signed-off-by: Karen Feng <[email protected]>

* rename test files

Signed-off-by: Karen Feng <[email protected]>

* Add license headers (#21)

* headers

* executable

* fix template rendering

* yapf

* add header to template

* add header to template

Signed-off-by: Henry D <[email protected]>

Co-authored-by: Kiavash Kianfar <[email protected]>
Co-authored-by: Karen Feng <[email protected]>
Co-authored-by: Leland <[email protected]>
Co-authored-by: Leland Barnard <[email protected]>
henrydavidge added a commit that referenced this pull request Jun 22, 2020
* Add Leland's demo notebook

* block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2)

* blocks

Signed-off-by: kianfar77 <[email protected]>

* test vcf

Signed-off-by: kianfar77 <[email protected]>

* transformer

Signed-off-by: kianfar77 <[email protected]>

* remove extra

Signed-off-by: kianfar77 <[email protected]>

* refactor and conform with ridge namings

Signed-off-by: kianfar77 <[email protected]>

* test

Signed-off-by: kianfar77 <[email protected]>

* test files

Signed-off-by: kianfar77 <[email protected]>

* remove extra file

Signed-off-by: kianfar77 <[email protected]>

* sort_key

Signed-off-by: kianfar77 <[email protected]>

* feat: ridge models for wgr added (#1)

* feat: ridge models for wgr added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Doc strings added for levels/functions.py
Some typos fixed in ridge_model.py
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* ridge_model and RidgeReducer unit tests added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* RidgeRegression unit tests added
test data README added
ridge_udfs.py docstrings added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Changes made to accessing the sample ID map and more docstrings

The map_normal_eqn and score_models functions previously expected the
sample IDs for a given sample block to be found in the Pandas DataFrame,
which mean we had to join them on before the .groupBy().apply().  These
functions now expect the sample block to sample IDs mapping to be
provided separately as a dict, so that the join is no longer required.
RidgeReducer and RidgeRegression APIs remain unchanged.

docstrings have been added for RidgeReducer and RidgeRegression classes.

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Refactored object names and comments to reflect new terminology

Where 'block' was previously used to refer to the set of columns in a
block, we now use 'header_block'
Where 'group' was previously used to refer to the set of samples in a
block, we now use 'sample_block'

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* address comments

Signed-off-by: Karen Feng <[email protected]>

* Test fixup

Signed-off-by: Karen Feng <[email protected]>

* Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching

Signed-off-by: Karen Feng <[email protected]>

* PyArrow 0.15.1 only with PySpark 3

Signed-off-by: Karen Feng <[email protected]>

* Don't use toPandas()

Signed-off-by: Karen Feng <[email protected]>

* Upgrade pyarrow

Signed-off-by: Karen Feng <[email protected]>

* Only register once

Signed-off-by: Karen Feng <[email protected]>

* Minimize memory usage

Signed-off-by: Karen Feng <[email protected]>

* Select before head

Signed-off-by: Karen Feng <[email protected]>

* set up/tear down

Signed-off-by: Karen Feng <[email protected]>

* Try limiting pyspark memory

Signed-off-by: Karen Feng <[email protected]>

* No teardown

Signed-off-by: Karen Feng <[email protected]>

* Extend timeout

Signed-off-by: Karen Feng <[email protected]>

* Simplify ordering logic in levels code (#7)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* start changing for readability

* use input label ordering

* rename create_row_indexer

* undo column sort

* change reduce

Signed-off-by: Henry D <[email protected]>

* further simplify reduce

* sorted alpha names

* remove ordering

* comments

Signed-off-by: Henry D <[email protected]>

* Set arrow env var in build

Signed-off-by: Henry D <[email protected]>

* faster sort

* add test file

* undo test data change

* >=

* formatting

* empty

Co-authored-by: Karen Feng <[email protected]>

* Limit Spark memory conf in tests (#9)

* yapf

Signed-off-by: Karen Feng <[email protected]>

* yapf transform

Signed-off-by: Karen Feng <[email protected]>

* Set driver memory

Signed-off-by: Karen Feng <[email protected]>

* Try changing spark mem

Signed-off-by: Karen Feng <[email protected]>

* match java tests

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* remove driver memory flag

Signed-off-by: Karen Feng <[email protected]>

* Improve partitioning in block_variants_and_samples transformer (#11)

Signed-off-by: kianfar77 <[email protected]>

* Remove unnecessary header_block grouping (#10)

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* Create sample ID blocking helper functions (#12)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* simplify tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* yapf

Signed-off-by: Karen Feng <[email protected]>

* index map compat

Signed-off-by: Karen Feng <[email protected]>

* Add docs

Signed-off-by: Karen Feng <[email protected]>

* Add more tests

Signed-off-by: Karen Feng <[email protected]>

* pass args as ints

Signed-off-by: Karen Feng <[email protected]>

* Don't roll our own splitter

Signed-off-by: Karen Feng <[email protected]>

* rename sample_index to sample_blocks

Signed-off-by: Karen Feng <[email protected]>

* Add type-checking to WGR APIs (#14)

* Add type-checking to APIs

Signed-off-by: Karen Feng <[email protected]>

* Check valid alphas

Signed-off-by: Karen Feng <[email protected]>

* check 0 sig

Signed-off-by: Karen Feng <[email protected]>

* Add to install_requires list

Signed-off-by: Karen Feng <[email protected]>

* cleanup comments

Signed-off-by: Karen Feng <[email protected]>

* Add covariate support (#13)

* Added necessary modifications to accomodate covariates in model fitting.

The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X.  This PR makes numerous changes to accomodate covariate matrix C.

Adding covariates required the following breaking changes to the APIs:
 * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform():
   * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf)
   * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf)

Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument.

Two new tests have been added to test_ridge_regression.py to test run modes with covariates:
 * test_ridge_reducer_transform_with_cov
 * test_two_level_regression_with_cov

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Cleaned up one unnecessary Pandas import
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Small changes for clarity and consistence with the rest of the code.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Forgot one usage of coalesce
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Added a couple of comments to explain logic and replaced usages of .values with .array
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Fixed one instance of the change .values -> .array where it was made in error.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Typo in test_ridge_regression.py.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Style auto-updates with yapfAll
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

Co-authored-by: Leland Barnard <[email protected]>
Co-authored-by: Karen Feng <[email protected]>

* Flatten estimated phenotypes (#15)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Clean up tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Order to match labeldf

Signed-off-by: Karen Feng <[email protected]>

* Check we tie-break

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* test var name

Signed-off-by: Karen Feng <[email protected]>

* clean up tests

Signed-off-by: Karen Feng <[email protected]>

* Clean up docs

Signed-off-by: Karen Feng <[email protected]>

* Add fit_transform function to models (#17)

Signed-off-by: Karen Feng <[email protected]>

* support alpha inference

Signed-off-by: Karen Feng <[email protected]>

* test fixup

Signed-off-by: Karen Feng <[email protected]>

* more test fixup

Signed-off-by: Karen Feng <[email protected]>

* test fixups

Signed-off-by: Karen Feng <[email protected]>

* sub-sample

Signed-off-by: Karen Feng <[email protected]>

* test fixup

Signed-off-by: Karen Feng <[email protected]>

* address comments - only infer alphas during fit

Signed-off-by: Karen Feng <[email protected]>

* exception varies

Signed-off-by: Karen Feng <[email protected]>

* Rename levels (#20)

* Rename levels to wgr

Signed-off-by: Karen Feng <[email protected]>

* rename test files

Signed-off-by: Karen Feng <[email protected]>

* Errors vary by Spark version

Signed-off-by: Karen Feng <[email protected]>

* Add license headers (#21)

* headers

* executable

* fix template rendering

* yapf

Co-authored-by: Kiavash Kianfar <[email protected]>
Co-authored-by: Karen Feng <[email protected]>
Co-authored-by: Leland <[email protected]>
Co-authored-by: Leland Barnard <[email protected]>
karenfeng added a commit that referenced this pull request Jun 23, 2020
* Add Leland's demo notebook

* block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2)

* blocks

Signed-off-by: kianfar77 <[email protected]>

* test vcf

Signed-off-by: kianfar77 <[email protected]>

* transformer

Signed-off-by: kianfar77 <[email protected]>

* remove extra

Signed-off-by: kianfar77 <[email protected]>

* refactor and conform with ridge namings

Signed-off-by: kianfar77 <[email protected]>

* test

Signed-off-by: kianfar77 <[email protected]>

* test files

Signed-off-by: kianfar77 <[email protected]>

* remove extra file

Signed-off-by: kianfar77 <[email protected]>

* sort_key

Signed-off-by: kianfar77 <[email protected]>

* feat: ridge models for wgr added (#1)

* feat: ridge models for wgr added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Doc strings added for levels/functions.py
Some typos fixed in ridge_model.py
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* ridge_model and RidgeReducer unit tests added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* RidgeRegression unit tests added
test data README added
ridge_udfs.py docstrings added
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Changes made to accessing the sample ID map and more docstrings

The map_normal_eqn and score_models functions previously expected the
sample IDs for a given sample block to be found in the Pandas DataFrame,
which mean we had to join them on before the .groupBy().apply().  These
functions now expect the sample block to sample IDs mapping to be
provided separately as a dict, so that the join is no longer required.
RidgeReducer and RidgeRegression APIs remain unchanged.

docstrings have been added for RidgeReducer and RidgeRegression classes.

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Refactored object names and comments to reflect new terminology

Where 'block' was previously used to refer to the set of columns in a
block, we now use 'header_block'
Where 'group' was previously used to refer to the set of samples in a
block, we now use 'sample_block'

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* address comments

Signed-off-by: Karen Feng <[email protected]>

* Test fixup

Signed-off-by: Karen Feng <[email protected]>

* Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching

Signed-off-by: Karen Feng <[email protected]>

* PyArrow 0.15.1 only with PySpark 3

Signed-off-by: Karen Feng <[email protected]>

* Don't use toPandas()

Signed-off-by: Karen Feng <[email protected]>

* Upgrade pyarrow

Signed-off-by: Karen Feng <[email protected]>

* Only register once

Signed-off-by: Karen Feng <[email protected]>

* Minimize memory usage

Signed-off-by: Karen Feng <[email protected]>

* Select before head

Signed-off-by: Karen Feng <[email protected]>

* set up/tear down

Signed-off-by: Karen Feng <[email protected]>

* Try limiting pyspark memory

Signed-off-by: Karen Feng <[email protected]>

* No teardown

Signed-off-by: Karen Feng <[email protected]>

* Extend timeout

Signed-off-by: Karen Feng <[email protected]>

* Simplify ordering logic in levels code (#7)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* existing tests pass

Signed-off-by: Karen Feng <[email protected]>

* rename file

Signed-off-by: Karen Feng <[email protected]>

* Add compat test

Signed-off-by: Karen Feng <[email protected]>

* scalafmt

Signed-off-by: Karen Feng <[email protected]>

* collect minimal columns

Signed-off-by: Karen Feng <[email protected]>

* start changing for readability

* use input label ordering

* rename create_row_indexer

* undo column sort

* change reduce

Signed-off-by: Henry D <[email protected]>

* further simplify reduce

* sorted alpha names

* remove ordering

* comments

Signed-off-by: Henry D <[email protected]>

* Set arrow env var in build

Signed-off-by: Henry D <[email protected]>

* faster sort

* add test file

* undo test data change

* >=

* formatting

* empty

Co-authored-by: Karen Feng <[email protected]>

* Limit Spark memory conf in tests (#9)

* yapf

Signed-off-by: Karen Feng <[email protected]>

* yapf transform

Signed-off-by: Karen Feng <[email protected]>

* Set driver memory

Signed-off-by: Karen Feng <[email protected]>

* Try changing spark mem

Signed-off-by: Karen Feng <[email protected]>

* match java tests

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* remove driver memory flag

Signed-off-by: Karen Feng <[email protected]>

* Improve partitioning in block_variants_and_samples transformer (#11)

Signed-off-by: kianfar77 <[email protected]>

* Remove unnecessary header_block grouping (#10)

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* Create sample ID blocking helper functions (#12)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* whoops

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* simplify tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* yapf

Signed-off-by: Karen Feng <[email protected]>

* index map compat

Signed-off-by: Karen Feng <[email protected]>

* Add docs

Signed-off-by: Karen Feng <[email protected]>

* Add more tests

Signed-off-by: Karen Feng <[email protected]>

* pass args as ints

Signed-off-by: Karen Feng <[email protected]>

* Don't roll our own splitter

Signed-off-by: Karen Feng <[email protected]>

* rename sample_index to sample_blocks

Signed-off-by: Karen Feng <[email protected]>

* Add type-checking to WGR APIs (#14)

* Add type-checking to APIs

Signed-off-by: Karen Feng <[email protected]>

* Check valid alphas

Signed-off-by: Karen Feng <[email protected]>

* check 0 sig

Signed-off-by: Karen Feng <[email protected]>

* Add to install_requires list

Signed-off-by: Karen Feng <[email protected]>

* cleanup comments

Signed-off-by: Karen Feng <[email protected]>

* Add covariate support (#13)

* Added necessary modifications to accomodate covariates in model fitting.

The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X.  This PR makes numerous changes to accomodate covariate matrix C.

Adding covariates required the following breaking changes to the APIs:
 * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform():
   * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf)
   * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf)

Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument.

Two new tests have been added to test_ridge_regression.py to test run modes with covariates:
 * test_ridge_reducer_transform_with_cov
 * test_two_level_regression_with_cov

Signed-off-by: Leland Barnard ([email protected])
Signed-off-by: Leland Barnard <[email protected]>

* Cleaned up one unnecessary Pandas import
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Small changes for clarity and consistence with the rest of the code.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Forgot one usage of coalesce
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Added a couple of comments to explain logic and replaced usages of .values with .array
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Fixed one instance of the change .values -> .array where it was made in error.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Typo in test_ridge_regression.py.
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

* Style auto-updates with yapfAll
Signed-off-by: Leland Barnard ([email protected])

Signed-off-by: Leland Barnard <[email protected]>

Co-authored-by: Leland Barnard <[email protected]>
Co-authored-by: Karen Feng <[email protected]>

* Flatten estimated phenotypes (#15)

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Clean up tests

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Order to match labeldf

Signed-off-by: Karen Feng <[email protected]>

* Check we tie-break

Signed-off-by: Karen Feng <[email protected]>

* cleanup

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* test var name

Signed-off-by: Karen Feng <[email protected]>

* clean up tests

Signed-off-by: Karen Feng <[email protected]>

* Clean up docs

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* tests

Signed-off-by: Karen Feng <[email protected]>

* remove accidental files

Signed-off-by: Karen Feng <[email protected]>

* Add fit_transform function to models (#17)

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* Rename levels (#20)

* Rename levels to wgr

Signed-off-by: Karen Feng <[email protected]>

* rename test files

Signed-off-by: Karen Feng <[email protected]>

* Add license headers (#21)

* headers

* executable

* fix template rendering

* yapf

* WIP

Signed-off-by: Karen Feng <[email protected]>

* WIP

Signed-off-by: Karen Feng <[email protected]>

* More work

Signed-off-by: Karen Feng <[email protected]>

* More cleanup

Signed-off-by: Karen Feng <[email protected]>

* Fix docs tests

Signed-off-by: Karen Feng <[email protected]>

* address comments

Signed-off-by: Karen Feng <[email protected]>

* fix regression fit description

Signed-off-by: Karen Feng <[email protected]>

* fix capitalization

Signed-off-by: Karen Feng <[email protected]>

* address some comments

Signed-off-by: Karen Feng <[email protected]>

* more cleanup

Signed-off-by: Karen Feng <[email protected]>

* More cleanup

Signed-off-by: Karen Feng <[email protected]>

* add notebook

Signed-off-by: Karen Feng <[email protected]>

* update notebook

Signed-off-by: Karen Feng <[email protected]>

Co-authored-by: Henry D <[email protected]>
Co-authored-by: Kiavash Kianfar <[email protected]>
Co-authored-by: Leland <[email protected]>
Co-authored-by: Leland Barnard <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants