From 02e43f7eaf29eb6b3aacd5dc76d5f8783ee0af4e Mon Sep 17 00:00:00 2001 From: Steffen Schneider Date: Wed, 3 Jan 2024 00:32:49 +0100 Subject: [PATCH] Apply updated license headers (#118) Apply license headers (#118) Co-authored-by: CEBRA <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Steffen Co-authored-by: MMathis --- cebra/__init__.py | 20 ++++-- cebra/__main__.py | 20 ++++-- cebra/config.py | 20 ++++-- cebra/data/__init__.py | 20 ++++-- cebra/data/assets.py | 20 ++++-- cebra/data/base.py | 20 ++++-- cebra/data/datasets.py | 20 ++++-- cebra/data/datatypes.py | 20 ++++-- cebra/data/helper.py | 20 ++++-- cebra/data/load.py | 20 ++++-- cebra/data/multi_session.py | 20 ++++-- cebra/data/single_session.py | 20 ++++-- cebra/datasets/__init__.py | 20 ++++-- cebra/datasets/allen/__init__.py | 20 ++++-- cebra/datasets/allen/ca_movie.py | 20 ++++-- cebra/datasets/allen/ca_movie_decoding.py | 20 ++++-- cebra/datasets/allen/combined.py | 20 ++++-- cebra/datasets/allen/make_neuropixel.py | 20 ++++-- cebra/datasets/allen/neuropixel_movie.py | 20 ++++-- .../allen/neuropixel_movie_decoding.py | 20 ++++-- cebra/datasets/allen/single_session_ca.py | 20 ++++-- cebra/datasets/demo.py | 20 ++++-- cebra/datasets/gaussian_mixture.py | 20 ++++-- cebra/datasets/generate_synthetic_data.py | 20 ++++-- cebra/datasets/hippocampus.py | 20 ++++-- cebra/datasets/make_neuropixel.py | 20 ++++-- cebra/datasets/monkey_reaching.py | 20 ++++-- cebra/datasets/poisson.py | 20 ++++-- cebra/datasets/save_dataset.py | 20 ++++-- cebra/datasets/synthetic_data.py | 20 ++++-- cebra/distributions/__init__.py | 20 ++++-- cebra/distributions/base.py | 20 ++++-- cebra/distributions/continuous.py | 20 ++++-- cebra/distributions/discrete.py | 20 ++++-- cebra/distributions/index.py | 20 ++++-- cebra/distributions/mixed.py | 20 ++++-- cebra/distributions/multisession.py | 20 ++++-- cebra/grid_search.py | 20 ++++-- cebra/helper.py | 20 ++++-- cebra/integrations/__init__.py | 20 ++++-- cebra/integrations/deeplabcut.py | 20 ++++-- cebra/integrations/matplotlib.py | 20 ++++-- cebra/integrations/plotly.py | 20 ++++-- cebra/integrations/sklearn/__init__.py | 20 ++++-- cebra/integrations/sklearn/cebra.py | 22 +++++-- cebra/integrations/sklearn/dataset.py | 20 ++++-- cebra/integrations/sklearn/decoder.py | 20 ++++-- cebra/integrations/sklearn/helpers.py | 20 ++++-- cebra/integrations/sklearn/metrics.py | 20 ++++-- cebra/integrations/sklearn/utils.py | 20 ++++-- cebra/io.py | 22 +++++-- cebra/models/__init__.py | 20 ++++-- cebra/models/criterions.py | 20 ++++-- cebra/models/layers.py | 20 ++++-- cebra/models/model.py | 20 ++++-- cebra/models/multiobjective.py | 20 ++++-- cebra/models/projector.py | 20 ++++-- cebra/registry.py | 20 ++++-- cebra/solver/__init__.py | 20 ++++-- cebra/solver/base.py | 20 ++++-- cebra/solver/multi_session.py | 20 ++++-- cebra/solver/single_session.py | 20 ++++-- cebra/solver/supervised.py | 20 ++++-- cebra/solver/util.py | 20 ++++-- conda/cebra.yml | 20 ++++-- conda/cebra_paper.yml | 20 ++++-- conda/cebra_paper_m1.yml | 20 ++++-- docs/source/conf.py | 28 ++++---- tests/_util.py | 20 ++++-- tests/test_api.py | 20 ++++-- tests/test_benchmark.py | 20 ++++-- tests/test_cli.py | 20 ++++-- tests/test_criterions.py | 20 ++++-- tests/test_data_helper.py | 20 ++++-- tests/test_datasets.py | 20 ++++-- tests/test_demo.py | 20 ++++-- tests/test_distributions.py | 20 ++++-- tests/test_dlc.py | 20 ++++-- tests/test_grid_search.py | 20 ++++-- tests/test_index.py | 20 ++++-- tests/test_integration_train.py | 20 ++++-- tests/test_io.py | 20 ++++-- tests/test_load.py | 65 ++++++++++++------- tests/test_loader.py | 20 ++++-- tests/test_models.py | 20 ++++-- tests/test_plot.py | 20 ++++-- tests/test_plotly.py | 20 ++++-- tests/test_registry.py | 20 ++++-- tests/test_sklearn.py | 30 ++++++--- tests/test_sklearn_decoder.py | 20 ++++-- tests/test_sklearn_metrics.py | 20 ++++-- tests/test_solver.py | 20 ++++-- tests/test_usecases.py | 20 ++++-- 93 files changed, 1429 insertions(+), 498 deletions(-) diff --git a/cebra/__init__.py b/cebra/__init__.py index cd3e9bba..ff9bc33e 100644 --- a/cebra/__init__.py +++ b/cebra/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """CEBRA is a library for estimating Consistent Embeddings of high-dimensional Recordings using Auxiliary variables. It contains self-supervised learning algorithms implemented in diff --git a/cebra/__main__.py b/cebra/__main__.py index a018ecf6..6c7c18bf 100644 --- a/cebra/__main__.py +++ b/cebra/__main__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """CEBRA command line interface. diff --git a/cebra/config.py b/cebra/config.py index 5de60af0..ba6e3922 100644 --- a/cebra/config.py +++ b/cebra/config.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import argparse import json diff --git a/cebra/data/__init__.py b/cebra/data/__init__.py index 41d22334..ec753f18 100644 --- a/cebra/data/__init__.py +++ b/cebra/data/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Data loaders use distributions and indices to make samples available for training. diff --git a/cebra/data/assets.py b/cebra/data/assets.py index 977bfbcc..6b1f1daf 100644 --- a/cebra/data/assets.py +++ b/cebra/data/assets.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import hashlib diff --git a/cebra/data/base.py b/cebra/data/base.py index 7c27d9b0..d2ee47b5 100644 --- a/cebra/data/base.py +++ b/cebra/data/base.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Base classes for datasets and loaders.""" diff --git a/cebra/data/datasets.py b/cebra/data/datasets.py index aa0ad3db..8fc990b0 100644 --- a/cebra/data/datasets.py +++ b/cebra/data/datasets.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Pre-defined datasets.""" diff --git a/cebra/data/datatypes.py b/cebra/data/datatypes.py index 433030b8..11583909 100644 --- a/cebra/data/datatypes.py +++ b/cebra/data/datatypes.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import collections from typing import Tuple diff --git a/cebra/data/helper.py b/cebra/data/helper.py index cea5ebc4..c324a80f 100644 --- a/cebra/data/helper.py +++ b/cebra/data/helper.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import copy import warnings diff --git a/cebra/data/load.py b/cebra/data/load.py index b03c58a5..ddaf8ade 100644 --- a/cebra/data/load.py +++ b/cebra/data/load.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """A simple API for loading various data formats used with CEBRA. diff --git a/cebra/data/multi_session.py b/cebra/data/multi_session.py index f9f5fd66..8cd74286 100644 --- a/cebra/data/multi_session.py +++ b/cebra/data/multi_session.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Datasets and loaders for multi-session training.""" diff --git a/cebra/data/single_session.py b/cebra/data/single_session.py index 6090545c..c27b10f5 100644 --- a/cebra/data/single_session.py +++ b/cebra/data/single_session.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Datasets and loaders for single session training. diff --git a/cebra/datasets/__init__.py b/cebra/datasets/__init__.py index 1cce8ab9..76bfed3c 100644 --- a/cebra/datasets/__init__.py +++ b/cebra/datasets/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Pre-defined demo and benchmark datasets. diff --git a/cebra/datasets/allen/__init__.py b/cebra/datasets/allen/__init__.py index ab76fa1f..be43b053 100644 --- a/cebra/datasets/allen/__init__.py +++ b/cebra/datasets/allen/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Datasets from the Allen Database diff --git a/cebra/datasets/allen/ca_movie.py b/cebra/datasets/allen/ca_movie.py index 1ae3bfb9..f11e5e93 100644 --- a/cebra/datasets/allen/ca_movie.py +++ b/cebra/datasets/allen/ca_movie.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Allen pseudomouse Ca dataset. diff --git a/cebra/datasets/allen/ca_movie_decoding.py b/cebra/datasets/allen/ca_movie_decoding.py index 7ef8931d..9a8a6317 100644 --- a/cebra/datasets/allen/ca_movie_decoding.py +++ b/cebra/datasets/allen/ca_movie_decoding.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Allen pseudomouse Ca decoding dataset with train/test split. diff --git a/cebra/datasets/allen/combined.py b/cebra/datasets/allen/combined.py index 13c207f2..bfaca9b3 100644 --- a/cebra/datasets/allen/combined.py +++ b/cebra/datasets/allen/combined.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Joint Allen pseudomouse Ca/Neuropixel datasets. diff --git a/cebra/datasets/allen/make_neuropixel.py b/cebra/datasets/allen/make_neuropixel.py index b656eb08..5c0568b7 100644 --- a/cebra/datasets/allen/make_neuropixel.py +++ b/cebra/datasets/allen/make_neuropixel.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Generate pseudomouse Neuropixels data. diff --git a/cebra/datasets/allen/neuropixel_movie.py b/cebra/datasets/allen/neuropixel_movie.py index 2bad04fa..097f2105 100644 --- a/cebra/datasets/allen/neuropixel_movie.py +++ b/cebra/datasets/allen/neuropixel_movie.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Allen pseudomouse Neuropixels decoding dataset. diff --git a/cebra/datasets/allen/neuropixel_movie_decoding.py b/cebra/datasets/allen/neuropixel_movie_decoding.py index 74d0640a..2fbc4c51 100644 --- a/cebra/datasets/allen/neuropixel_movie_decoding.py +++ b/cebra/datasets/allen/neuropixel_movie_decoding.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Allen pseudomouse Neuropixels decoding dataset with train/test split. diff --git a/cebra/datasets/allen/single_session_ca.py b/cebra/datasets/allen/single_session_ca.py index c1fa949a..f207a1bc 100644 --- a/cebra/datasets/allen/single_session_ca.py +++ b/cebra/datasets/allen/single_session_ca.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Allen single mouse dataset. diff --git a/cebra/datasets/demo.py b/cebra/datasets/demo.py index b296e203..3e943b07 100644 --- a/cebra/datasets/demo.py +++ b/cebra/datasets/demo.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Demo datasets for testing CEBRA. diff --git a/cebra/datasets/gaussian_mixture.py b/cebra/datasets/gaussian_mixture.py index 1ab51bad..63bd1009 100644 --- a/cebra/datasets/gaussian_mixture.py +++ b/cebra/datasets/gaussian_mixture.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # from typing import Tuple diff --git a/cebra/datasets/generate_synthetic_data.py b/cebra/datasets/generate_synthetic_data.py index e6dd0802..0b75baf5 100644 --- a/cebra/datasets/generate_synthetic_data.py +++ b/cebra/datasets/generate_synthetic_data.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Generate synthetic datasets for benchmarking embedding quality. diff --git a/cebra/datasets/hippocampus.py b/cebra/datasets/hippocampus.py index 06db6daf..29962cb8 100644 --- a/cebra/datasets/hippocampus.py +++ b/cebra/datasets/hippocampus.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Rat hippocampus dataset diff --git a/cebra/datasets/make_neuropixel.py b/cebra/datasets/make_neuropixel.py index b7d200fc..27745ee3 100644 --- a/cebra/datasets/make_neuropixel.py +++ b/cebra/datasets/make_neuropixel.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Generate pseudomouse Neuropixels data. diff --git a/cebra/datasets/monkey_reaching.py b/cebra/datasets/monkey_reaching.py index 629cc4a5..d98fddae 100644 --- a/cebra/datasets/monkey_reaching.py +++ b/cebra/datasets/monkey_reaching.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Ephys neural and behavior data used for the monkey reaching experiment. diff --git a/cebra/datasets/poisson.py b/cebra/datasets/poisson.py index 6b907b01..e41d2236 100644 --- a/cebra/datasets/poisson.py +++ b/cebra/datasets/poisson.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Utilities for simulating spike counts for Poisson-like neuron models.""" diff --git a/cebra/datasets/save_dataset.py b/cebra/datasets/save_dataset.py index 1b941f05..f5e01a62 100644 --- a/cebra/datasets/save_dataset.py +++ b/cebra/datasets/save_dataset.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import argparse import os diff --git a/cebra/datasets/synthetic_data.py b/cebra/datasets/synthetic_data.py index 8b9c4cbf..9288a93d 100644 --- a/cebra/datasets/synthetic_data.py +++ b/cebra/datasets/synthetic_data.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import os diff --git a/cebra/distributions/__init__.py b/cebra/distributions/__init__.py index 56caedba..6985adbc 100644 --- a/cebra/distributions/__init__.py +++ b/cebra/distributions/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Distributions and indexing helper functions for training CEBRA models. diff --git a/cebra/distributions/base.py b/cebra/distributions/base.py index 94092e4e..990d7e79 100644 --- a/cebra/distributions/base.py +++ b/cebra/distributions/base.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Abstract base classes for distributions and indices. diff --git a/cebra/distributions/continuous.py b/cebra/distributions/continuous.py index 2a8d0337..c4235d48 100644 --- a/cebra/distributions/continuous.py +++ b/cebra/distributions/continuous.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Distributions for sampling from continuously indexed datasets.""" diff --git a/cebra/distributions/discrete.py b/cebra/distributions/discrete.py index dcb9085e..531dfc16 100644 --- a/cebra/distributions/discrete.py +++ b/cebra/distributions/discrete.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Discrete indices.""" diff --git a/cebra/distributions/index.py b/cebra/distributions/index.py index c8c5b091..0ee0959a 100644 --- a/cebra/distributions/index.py +++ b/cebra/distributions/index.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Index operations for conditional sampling. diff --git a/cebra/distributions/mixed.py b/cebra/distributions/mixed.py index 78def25f..14fb8a61 100644 --- a/cebra/distributions/mixed.py +++ b/cebra/distributions/mixed.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Distributions with a mix of continuous/discrete auxiliary variables. diff --git a/cebra/distributions/multisession.py b/cebra/distributions/multisession.py index af6fe0ad..08595d90 100644 --- a/cebra/distributions/multisession.py +++ b/cebra/distributions/multisession.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Continuous variable multi-session sampling.""" diff --git a/cebra/grid_search.py b/cebra/grid_search.py index cbc05c95..14337ac0 100644 --- a/cebra/grid_search.py +++ b/cebra/grid_search.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Utilities for performing a grid search across CEBRA models.""" diff --git a/cebra/helper.py b/cebra/helper.py index b76fd93e..2175e6ac 100644 --- a/cebra/helper.py +++ b/cebra/helper.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Collection of helper functions that did not fit into own modules.""" diff --git a/cebra/integrations/__init__.py b/cebra/integrations/__init__.py index fcbb713f..ed363fc4 100644 --- a/cebra/integrations/__init__.py +++ b/cebra/integrations/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Integration of CEBRA with common machine learning and visualization libraries. diff --git a/cebra/integrations/deeplabcut.py b/cebra/integrations/deeplabcut.py index 17955c35..c265b09a 100644 --- a/cebra/integrations/deeplabcut.py +++ b/cebra/integrations/deeplabcut.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Helper functions for training embeddings on DeepLabCut outputs.""" diff --git a/cebra/integrations/matplotlib.py b/cebra/integrations/matplotlib.py index 4fe9e358..b79deb66 100644 --- a/cebra/integrations/matplotlib.py +++ b/cebra/integrations/matplotlib.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Matplotlib interface to CEBRA.""" import abc diff --git a/cebra/integrations/plotly.py b/cebra/integrations/plotly.py index 8f0c2089..9dd4fa02 100644 --- a/cebra/integrations/plotly.py +++ b/cebra/integrations/plotly.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Plotly interface to CEBRA.""" from typing import Optional, Tuple, Union diff --git a/cebra/integrations/sklearn/__init__.py b/cebra/integrations/sklearn/__init__.py index 85367428..7f590af5 100644 --- a/cebra/integrations/sklearn/__init__.py +++ b/cebra/integrations/sklearn/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Scikit-Learn interface for CEBRA diff --git a/cebra/integrations/sklearn/cebra.py b/cebra/integrations/sklearn/cebra.py index 93804a99..077d3c47 100644 --- a/cebra/integrations/sklearn/cebra.py +++ b/cebra/integrations/sklearn/cebra.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Define the CEBRA model.""" @@ -1402,7 +1412,7 @@ def load(cls, Example: >>> import cebra - >>> import numpy as np + >>> import numpy as np >>> import tempfile >>> from pathlib import Path >>> tmp_file = Path(tempfile.gettempdir(), 'cebra.pt') diff --git a/cebra/integrations/sklearn/dataset.py b/cebra/integrations/sklearn/dataset.py index 6a5b0753..1bb635ec 100644 --- a/cebra/integrations/sklearn/dataset.py +++ b/cebra/integrations/sklearn/dataset.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Datasets to be used as part of the sklearn framework.""" diff --git a/cebra/integrations/sklearn/decoder.py b/cebra/integrations/sklearn/decoder.py index 9dc412e2..03b93238 100644 --- a/cebra/integrations/sklearn/decoder.py +++ b/cebra/integrations/sklearn/decoder.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Some decoders following the ``scikit-learn`` API.""" diff --git a/cebra/integrations/sklearn/helpers.py b/cebra/integrations/sklearn/helpers.py index 9f9d877d..06095c1e 100644 --- a/cebra/integrations/sklearn/helpers.py +++ b/cebra/integrations/sklearn/helpers.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # from typing import List, Tuple, Union diff --git a/cebra/integrations/sklearn/metrics.py b/cebra/integrations/sklearn/metrics.py index 7df34b69..9712d021 100644 --- a/cebra/integrations/sklearn/metrics.py +++ b/cebra/integrations/sklearn/metrics.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # from typing import Iterable, List, Literal, Optional, Tuple, Union diff --git a/cebra/integrations/sklearn/utils.py b/cebra/integrations/sklearn/utils.py index 6510c133..455213a3 100644 --- a/cebra/integrations/sklearn/utils.py +++ b/cebra/integrations/sklearn/utils.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import warnings diff --git a/cebra/io.py b/cebra/io.py index bb0e2136..2b32bcaa 100644 --- a/cebra/io.py +++ b/cebra/io.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Helper classes and functions for I/O functionality.""" @@ -218,7 +228,7 @@ class FileKeyValueDataset: >>> from pathlib import Path >>> tmp_file = Path(tempfile.gettempdir(), 'test.jl') >>> _ = joblib.dump({'foo' : 42}, tmp_file) - >>> data = cebra.io.FileKeyValueDataset(tmp_file) + >>> data = cebra.io.FileKeyValueDataset(tmp_file) >>> data.foo 42 diff --git a/cebra/models/__init__.py b/cebra/models/__init__.py index e80ed379..4dfad333 100644 --- a/cebra/models/__init__.py +++ b/cebra/models/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Pre-defined neural network model architectures diff --git a/cebra/models/criterions.py b/cebra/models/criterions.py index 42ebb7ad..8dbdc2b4 100644 --- a/cebra/models/criterions.py +++ b/cebra/models/criterions.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Criterions for contrastive learning diff --git a/cebra/models/layers.py b/cebra/models/layers.py index f3cae690..7c1c36e8 100644 --- a/cebra/models/layers.py +++ b/cebra/models/layers.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Neural network layers used for building cebra models. diff --git a/cebra/models/model.py b/cebra/models/model.py index f9567605..f4a5d862 100644 --- a/cebra/models/model.py +++ b/cebra/models/model.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Neural network models and criterions for training CEBRA models.""" import abc diff --git a/cebra/models/multiobjective.py b/cebra/models/multiobjective.py index f55bc207..da7f992e 100644 --- a/cebra/models/multiobjective.py +++ b/cebra/models/multiobjective.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Wrappers for using models with multiobjective solvers. diff --git a/cebra/models/projector.py b/cebra/models/projector.py index 172c2468..0c924296 100644 --- a/cebra/models/projector.py +++ b/cebra/models/projector.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Training CEBRA with projectors.""" diff --git a/cebra/registry.py b/cebra/registry.py index 359d1def..be9afbd0 100644 --- a/cebra/registry.py +++ b/cebra/registry.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """A simple registry for python modules. diff --git a/cebra/solver/__init__.py b/cebra/solver/__init__.py index b9d27168..12ad2f06 100644 --- a/cebra/solver/__init__.py +++ b/cebra/solver/__init__.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Variants of CEBRA solvers for single- and multi-session training. diff --git a/cebra/solver/base.py b/cebra/solver/base.py index 92103173..c350ba35 100644 --- a/cebra/solver/base.py +++ b/cebra/solver/base.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """This package contains abstract base classes for different solvers. diff --git a/cebra/solver/multi_session.py b/cebra/solver/multi_session.py index be40d6da..7f103708 100644 --- a/cebra/solver/multi_session.py +++ b/cebra/solver/multi_session.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Solver implementations for multi-session datasetes.""" diff --git a/cebra/solver/single_session.py b/cebra/solver/single_session.py index 4dc03dde..ded526e9 100644 --- a/cebra/solver/single_session.py +++ b/cebra/solver/single_session.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Single session solvers embed a single pair of time series.""" diff --git a/cebra/solver/supervised.py b/cebra/solver/supervised.py index 49812aef..471e601f 100644 --- a/cebra/solver/supervised.py +++ b/cebra/solver/supervised.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Solvers for supervised training diff --git a/cebra/solver/util.py b/cebra/solver/util.py index fa0cecb9..584eb0da 100644 --- a/cebra/solver/util.py +++ b/cebra/solver/util.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Utility functions for solvers and their training loops.""" diff --git a/conda/cebra.yml b/conda/cebra.yml index 7bc9eb7c..03a6371f 100644 --- a/conda/cebra.yml +++ b/conda/cebra.yml @@ -1,13 +1,23 @@ ## -## (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -## Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -## original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -## +## CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +## © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) ## Source code: ## https://github.com/AdaptiveMotorControlLab/CEBRA ## ## Please see LICENSE.md for the full license document: -## https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +## https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +## +## Licensed under the Apache License, Version 2.0 (the "License"); +## you may not use this file except in compliance with the License. +## You may obtain a copy of the License at +## +## http://www.apache.org/licenses/LICENSE-2.0 +## +## Unless required by applicable law or agreed to in writing, software +## distributed under the License is distributed on an "AS IS" BASIS, +## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +## See the License for the specific language governing permissions and +## limitations under the License. ## # This environment can be used to install cebra diff --git a/conda/cebra_paper.yml b/conda/cebra_paper.yml index e4217679..e7537756 100644 --- a/conda/cebra_paper.yml +++ b/conda/cebra_paper.yml @@ -1,13 +1,23 @@ ## -## (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -## Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -## original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -## +## CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +## © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) ## Source code: ## https://github.com/AdaptiveMotorControlLab/CEBRA ## ## Please see LICENSE.md for the full license document: -## https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +## https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +## +## Licensed under the Apache License, Version 2.0 (the "License"); +## you may not use this file except in compliance with the License. +## You may obtain a copy of the License at +## +## http://www.apache.org/licenses/LICENSE-2.0 +## +## Unless required by applicable law or agreed to in writing, software +## distributed under the License is distributed on an "AS IS" BASIS, +## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +## See the License for the specific language governing permissions and +## limitations under the License. ## # This environment is the full requirements to reproduce the first cebra paper (although we recommend using docker). diff --git a/conda/cebra_paper_m1.yml b/conda/cebra_paper_m1.yml index 16be7c2c..32256758 100644 --- a/conda/cebra_paper_m1.yml +++ b/conda/cebra_paper_m1.yml @@ -1,13 +1,23 @@ ## -## (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -## Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -## original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -## +## CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +## © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) ## Source code: ## https://github.com/AdaptiveMotorControlLab/CEBRA ## ## Please see LICENSE.md for the full license document: -## https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +## https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +## +## Licensed under the Apache License, Version 2.0 (the "License"); +## you may not use this file except in compliance with the License. +## You may obtain a copy of the License at +## +## http://www.apache.org/licenses/LICENSE-2.0 +## +## Unless required by applicable law or agreed to in writing, software +## distributed under the License is distributed on an "AS IS" BASIS, +## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +## See the License for the specific language governing permissions and +## limitations under the License. ## #NOTE: if you need to install piVAE, then you need to install tensorflow for apple. This is NOT required for cebra-only use. diff --git a/docs/source/conf.py b/docs/source/conf.py index f868c185..c35ed9a5 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # # Configuration file for the Sphinx documentation builder. @@ -111,13 +121,7 @@ def get_years(start_year=2021): autodoc_member_order = "bysource" autodoc_mock_imports = [ - "torch", - "nlb_tools", - "tqdm", - "h5py", - "pandas", - "matplotlib", - "plotly" + "torch", "nlb_tools", "tqdm", "h5py", "pandas", "matplotlib", "plotly" ] # autodoc_typehints = "none" diff --git a/tests/_util.py b/tests/_util.py index 404e9eff..b4a0e07d 100644 --- a/tests/_util.py +++ b/tests/_util.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import collections.abc as collections_abc diff --git a/tests/test_api.py b/tests/test_api.py index 17317ddf..bc279cbd 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # def test_api(): import cebra.distributions diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py index 7a2e24f7..4fd51b07 100644 --- a/tests/test_benchmark.py +++ b/tests/test_benchmark.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import numpy as np import pytest diff --git a/tests/test_cli.py b/tests/test_cli.py index fe1f8082..41e67f42 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import argparse diff --git a/tests/test_criterions.py b/tests/test_criterions.py index 1a982860..24c47751 100644 --- a/tests/test_criterions.py +++ b/tests/test_criterions.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import numpy as np import pytest diff --git a/tests/test_data_helper.py b/tests/test_data_helper.py index 5154197b..a3d1e774 100644 --- a/tests/test_data_helper.py +++ b/tests/test_data_helper.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import numpy as np import pytest diff --git a/tests/test_datasets.py b/tests/test_datasets.py index a1b1558d..bc65d4b4 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import os import pathlib diff --git a/tests/test_demo.py b/tests/test_demo.py index 48753882..4f0f146c 100644 --- a/tests/test_demo.py +++ b/tests/test_demo.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import glob import re diff --git a/tests/test_distributions.py b/tests/test_distributions.py index 68374f4d..f47fc630 100644 --- a/tests/test_distributions.py +++ b/tests/test_distributions.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import functools from typing import Literal, Optional diff --git a/tests/test_dlc.py b/tests/test_dlc.py index bbac1d89..397cc4ba 100644 --- a/tests/test_dlc.py +++ b/tests/test_dlc.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import tempfile diff --git a/tests/test_grid_search.py b/tests/test_grid_search.py index 33458c8c..3f88ba12 100644 --- a/tests/test_grid_search.py +++ b/tests/test_grid_search.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import numpy as np import pytest diff --git a/tests/test_index.py b/tests/test_index.py index be0566ba..8e49cc35 100644 --- a/tests/test_index.py +++ b/tests/test_index.py @@ -1,11 +1,21 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # diff --git a/tests/test_integration_train.py b/tests/test_integration_train.py index 839d63cd..6e25f116 100644 --- a/tests/test_integration_train.py +++ b/tests/test_integration_train.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import itertools from typing import List diff --git a/tests/test_io.py b/tests/test_io.py index 22dfed93..541948ef 100644 --- a/tests/test_io.py +++ b/tests/test_io.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import _util import torch diff --git a/tests/test_load.py b/tests/test_load.py index be797ca8..6f62dc92 100644 --- a/tests/test_load.py +++ b/tests/test_load.py @@ -1,16 +1,28 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # +import itertools import pathlib import pickle +import platform import tempfile import unittest from unittest.mock import patch @@ -24,8 +36,6 @@ import pytest import scipy.io import torch -import itertools -import platform import cebra.data.load as cebra_load @@ -54,14 +64,13 @@ def _register(f): # TODO: test loading for file without extension # __test_functions.append(f) # with all possible extensions - __test_functions.extend([ - (lambda filename, dtype: f(filename + "." + file_ending, dtype=dtype), file_ending) - for file_ending in file_endings - ]) + __test_functions.extend([(lambda filename, dtype: f( + filename + "." + file_ending, dtype=dtype), file_ending) + for file_ending in file_endings]) if len(requires) > 0: __test_functions_module_not_found.extend([ - (requires, lambda filename: filename + "." + file_ending, - lambda filename, dtype: f(filename + "." + file_ending, dtype=dtype)) + (requires, lambda filename: filename + "." + file_ending, lambda + filename, dtype: f(filename + "." + file_ending, dtype=dtype)) for file_ending in file_endings ]) return f @@ -771,20 +780,23 @@ def generate_mat_new_no_array(filename, dtype): test_functions = list(itertools.product(__test_functions, [np.int32, np.int64])) test_functions = [(*t, x) for t, x in test_functions] -@pytest.mark.parametrize("save_data,file_ending,dtype", test_functions) + + +@pytest.mark.parametrize("save_data,file_ending,dtype", test_functions) def test_load(save_data, file_ending, dtype): with tempfile.NamedTemporaryFile() as tf: filename = tf.name # name, without extension - + if file_ending in ("csv", "xls", "xlsx", "xlsm"): if dtype == np.int32: - pytest.skip("Skipping test. For CSV, XLS, XLSX, and XLM file formats, " - "the integer loading data type is always int64, regardless of the " - "data type it was saved with. This can lead to compatibility issues, " - "especially on Windows. To ensure accurate testing, we only perform " - "tests with int64 data type for these formats, and we skip the test " - "cases involving int32.") - + pytest.skip( + "Skipping test. For CSV, XLS, XLSX, and XLM file formats, " + "the integer loading data type is always int64, regardless of the " + "data type it was saved with. This can lead to compatibility issues, " + "especially on Windows. To ensure accurate testing, we only perform " + "tests with int64 data type for these formats, and we skip the test " + "cases involving int32.") + # create data, save it, load it saved_array, loaded_array = save_data(filename, dtype=dtype) @@ -803,8 +815,13 @@ def test_load_error(save_data): save_data(filename) -test_functions_module_not_found = list(itertools.product(__test_functions_module_not_found, [np.int32, np.int64])) -test_functions_module_not_found = [(*t, x) for t, x in test_functions_module_not_found] +test_functions_module_not_found = list( + itertools.product(__test_functions_module_not_found, [np.int32, np.int64])) +test_functions_module_not_found = [ + (*t, x) for t, x in test_functions_module_not_found +] + + @pytest.mark.parametrize("module_names,get_path,save_data,dtype", test_functions_module_not_found) def test_module_not_installed(module_names, get_path, save_data, dtype): diff --git a/tests/test_loader.py b/tests/test_loader.py index 3d73573e..c7548338 100644 --- a/tests/test_loader.py +++ b/tests/test_loader.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import pytest import torch diff --git a/tests/test_models.py b/tests/test_models.py index 80fecdb1..2a6e4812 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import itertools diff --git a/tests/test_plot.py b/tests/test_plot.py index eef99f64..3f44d887 100644 --- a/tests/test_plot.py +++ b/tests/test_plot.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import copy import itertools diff --git a/tests/test_plotly.py b/tests/test_plotly.py index 4f0a926b..90fb64ed 100644 --- a/tests/test_plotly.py +++ b/tests/test_plotly.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import matplotlib import numpy as np diff --git a/tests/test_registry.py b/tests/test_registry.py index c0711d53..69e04f38 100644 --- a/tests/test_registry.py +++ b/tests/test_registry.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import types diff --git a/tests/test_sklearn.py b/tests/test_sklearn.py index 36ffd2fd..649a7c93 100644 --- a/tests/test_sklearn.py +++ b/tests/test_sklearn.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import itertools import tempfile @@ -703,19 +713,19 @@ def test_sklearn_adapt_multisession(model_architecture, device): batch_size=42, verbose=True, ) - + # example dataset Xs = [np.random.uniform(0, 1, (1000, 50)) for i in range(3)] ys = [np.random.uniform(0, 1, (1000, 5)) for i in range(3)] - + X_new = np.random.uniform(0, 1, (1000, 50)) y_new = np.random.uniform(0, 1, (1000, 5)) - + cebra_model.fit(Xs, ys) - + with pytest.raises(NotImplementedError, match=".*multisession.*"): cebra_model.fit(X_new, y_new, adapt=True) - + @_util.parametrize_slow( arg_names="model_architecture,device,pad_before_transform", diff --git a/tests/test_sklearn_decoder.py b/tests/test_sklearn_decoder.py index 6b4659a6..4538d9b4 100644 --- a/tests/test_sklearn_decoder.py +++ b/tests/test_sklearn_decoder.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import numpy as np import pytest diff --git a/tests/test_sklearn_metrics.py b/tests/test_sklearn_metrics.py index 3764b498..58e12010 100644 --- a/tests/test_sklearn_metrics.py +++ b/tests/test_sklearn_metrics.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import math diff --git a/tests/test_solver.py b/tests/test_solver.py index 443f26c5..46efd319 100644 --- a/tests/test_solver.py +++ b/tests/test_solver.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # import itertools diff --git a/tests/test_usecases.py b/tests/test_usecases.py index 50ecb412..22195bd8 100644 --- a/tests/test_usecases.py +++ b/tests/test_usecases.py @@ -1,13 +1,23 @@ # -# (c) All rights reserved. ECOLE POLYTECHNIQUE FÉDÉRALE DE LAUSANNE, -# Switzerland, Laboratory of Prof. Mackenzie W. Mathis (UPMWMATHIS) and -# original authors: Steffen Schneider, Jin H Lee, Mackenzie W Mathis. 2023. -# +# CEBRA: Consistent EmBeddings of high-dimensional Recordings using Auxiliary variables +# © Mackenzie W. Mathis & Steffen Schneider (v0.4.0+) # Source code: # https://github.com/AdaptiveMotorControlLab/CEBRA # # Please see LICENSE.md for the full license document: -# https://github.com/AdaptiveMotorControlLab/CEBRA/LICENSE.md +# https://github.com/AdaptiveMotorControlLab/CEBRA/blob/main/LICENSE.md +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. # """Integrations tests using various usecases via the sklearn API.