diff --git a/calculate_statistics.py b/calculate_statistics.py index ac0ef48..93d39d3 100644 --- a/calculate_statistics.py +++ b/calculate_statistics.py @@ -19,6 +19,22 @@ """ +""" + * Copyright 2024 Comcast Cable Communications Management, LLC + * + * 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 check_element_wise_equality(lists): counter = 0 diff --git a/helper_functions.py b/helper_functions.py index 705cc20..765cb3b 100644 --- a/helper_functions.py +++ b/helper_functions.py @@ -11,6 +11,22 @@ Contains helper functions. """ +""" + * Copyright 2024 Comcast Cable Communications Management, LLC + * + * 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 parse_file_name(file_name: str): """Splits file_name into model_name, task and experiment_seed diff --git a/main.py b/main.py index de1d5a8..e141c53 100644 --- a/main.py +++ b/main.py @@ -14,6 +14,22 @@ """Top level main function to run LLMs.""" +""" + * Copyright 2024 Comcast Cable Communications Management, LLC + * + * 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. +""" + parser = argparse.ArgumentParser(description="") parser.add_argument("--task", type=str) parser.add_argument("--model", type=str) diff --git a/postprocess_responses.py b/postprocess_responses.py index 96843f5..80824f1 100644 --- a/postprocess_responses.py +++ b/postprocess_responses.py @@ -9,6 +9,22 @@ """Takes a llm_output.csv file from an LLM run and parses out the answer and saves as parsed_llm_output.csv""" +""" + * Copyright 2024 Comcast Cable Communications Management, LLC + * + * 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 process_raw_response(row): response = row["raw_response"]